Searched refs:GenerateIndices (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_nd_op_cpu_impl.h | 54 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE bool GenerateIndices( in GenerateIndices() function
73 const bool out_of_bounds = GenerateIndices(loc, &ix); in operator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1086 void GenerateIndices(const absl::Span<const int64> dst_lengths, in GenerateIndices() function
1167 GenerateIndices(fft_lengths, fft_strides, input_lengths, input_strides, in CopyDataFromInput()
1215 GenerateIndices(output_lengths, output_strides, fft_lengths, fft_strides, in CopyDataToOutput()
1440 GenerateIndices(output_lengths, output_strides, input_lengths, in HandleFft()
|