Searched defs:indices_out (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dtopk_op.cc81 Tensor* indices_out = nullptr; variable
83 context->allocate_output(1, output_shape, &indices_out));
89 auto indices = indices_out->flat_inner_dims<int32>();
H A Dsparse_cross_op.cc143 Tensor* indices_out, Tensor* values_out)
145 indices_out_(indices_out),
320 Tensor* indices_out; variable
325 CreateOutputTensors(columns, batch_size, context, &indices_out, &values_out,
329 output_start_indices, indices_out, values_out);
506 int64 batch_size, OpKernelContext* context, Tensor** indices_out,
523 0, TensorShape({cross_count_total, 2}), indices_out));
142 OutputUpdater(const std::vector<int64>& output_start_indices, Tensor* indices_out, Tensor* values_out) argument
503 CreateOutputTensors( const std::vector<std::unique_ptr<ColumnInterface<InternalType>>>& columns, int64 batch_size, OpKernelContext* context, Tensor** indices_out, Tensor** values_out, Tensor** shape_out, std::vector<int64>* output_start_indices) argument
H A Ddynamic_partition_op_gpu.cu.cc21 // partitions_out and indices_out, with partitions_out sorted.
30 // 5. Finally, we use indices_out and the gather functor to collect the output.
31 // This works, because for each interval of i-values, indices_out points
197 Tensor* partitions_out, Tensor* indices_out,
207 c, c->allocate_temp(DT_INT32, TensorShape({N}), indices_out), done);
266 Tensor indices_out; local
268 // Also sort the info in partitions and output it in indices_out,
270 this->CountAndSortParts(c, &partitions, &partition_count, &indices_out,
303 auto wrapped_callback = [this, c, &data, &partitions, indices_out,
313 this->GatherSlices(c, &data, &indices_out,
196 AllocateTempSpace(OpKernelContext* c, int32 N, Tensor* indices_in, Tensor* partitions_out, Tensor* indices_out, DoneCallback done) argument
323 RadixSort(OpKernelContext* c, const Tensor* partitions, Tensor* indices_in, Tensor* partitions_out, Tensor* indices_out, DoneCallback done) argument
357 CountAndSortParts(OpKernelContext* c, const Tensor* partitions, Tensor* partition_count, Tensor* indices_out, DoneCallback done) argument
[all...]
/external/tensorflow/tensorflow/contrib/layers/kernels/
H A Dsparse_feature_cross_kernel.cc143 Tensor* indices_out, Tensor* values_out)
145 indices_out_(indices_out),
363 Tensor* indices_out; variable
368 CreateOutputTensors(columns, batch_size, context, &indices_out, &values_out,
372 updater(output_start_indices, indices_out, values_out);
549 int64 batch_size, OpKernelContext* context, Tensor** indices_out,
566 0, TensorShape({cross_count_total, 2}), indices_out));
142 OutputUpdater(const std::vector<int64>& output_start_indices, Tensor* indices_out, Tensor* values_out) argument
546 CreateOutputTensors( const std::vector<std::unique_ptr<ColumnInterface<InternalType>>>& columns, int64 batch_size, OpKernelContext* context, Tensor** indices_out, Tensor** values_out, Tensor** shape_out, std::vector<int64>* output_start_indices) argument

Completed in 141 milliseconds