Searched defs:input_indices (Results 1 – 12 of 12) sorted by relevance
34 CudaDeviceArrayStruct<int32> input_indices, in DynamicStitchKernel()54 const CudaDeviceArrayStruct<int>& input_indices, in DynamicStitchGPUImpl()
34 const Tensor& input_indices = context->input(1); in Compute() local
31 const Tensor& input_indices = context->input(0); in Compute() local
51 const Tensor* input_indices; in Compute() local143 const Tensor* input_indices; in Compute() local
168 const Tensor* input_indices; in Compute() local221 const Tensor* input_indices; in Compute() local
59 gtl::InlinedVector<int, 8> input_indices(num); in Compute() local
440 Tensor input_indices; in LaunchSortKernel() local
370 CodepointSpan input_indices = in JNI_METHOD() local391 const CodepointSpan input_indices = in JNI_METHOD() local
76 const Tensor& input_indices = context->input(4); in Compute() local
556 ShapeHandle input_indices = c->input(0); in __anone6e195411502() local
191 const std::vector<int>* input_indices, in FlattenInputs()
1320 tensorflow::gtl::ArraySlice<int64> input_indices = output_indices; in Replicate() local