Searched defs:indices_out (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | topk_op.cc | 81 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 D | sparse_cross_op.cc | 143 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 D | dynamic_partition_op_gpu.cu.cc | 21 // 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 D | sparse_feature_cross_kernel.cc | 143 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