Searched defs:indices_in (Results 1 – 3 of 3) sorted by relevance
71 const Tensor& indices_in = in Compute() local103 ::tensorflow::Status ValidateIndices(const Tensor& indices_in, in ValidateIndices()122 const Tensor& indices_in, const OpInputList& params_nested_splits_in, in MakeSplits()
193 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
46 const Tensor& indices_in = context->input(1); in Compute() local