Home
last modified time | relevance | path

Searched defs:indices_in (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dragged_gather_op.cc71 const Tensor& indices_in = in Compute() local
103 ::tensorflow::Status ValidateIndices(const Tensor& indices_in, in ValidateIndices()
122 const Tensor& indices_in, const OpInputList& params_nested_splits_in, in MakeSplits()
Ddynamic_partition_op_gpu.cu.cc193 void AllocateTempSpace(OpKernelContext* c, int32 N, Tensor* indices_in, in AllocateTempSpace()
324 Tensor* indices_in, Tensor* partitions_out, in RadixSort()
363 Tensor indices_in; in CountAndSortParts() local
Dgpu_prim_helpers_test.cu.cc46 const Tensor& indices_in = context->input(1); in Compute() local