Searched refs:candidate_input_indices (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 859 gtl::ArraySlice<int> candidate_input_indices, int output_index,
870 gtl::ArraySlice<int> candidate_input_indices, DataType type,
875 gtl::ArraySlice<int> candidate_input_indices, DataType type, in forward_input_or_allocate_temp() argument
877 return forward_input_or_allocate_temp(candidate_input_indices, type, shape, in forward_input_or_allocate_temp()
1586 gtl::ArraySlice<int> candidate_input_indices, int output_index, in forward_input_or_allocate_output() argument
1588 for (int input_index : candidate_input_indices) { in forward_input_or_allocate_output()
|
| D | op_kernel.cc | 565 gtl::ArraySlice<int> candidate_input_indices, DataType type, in forward_input_or_allocate_temp() argument
568 for (int input_index : candidate_input_indices) { in forward_input_or_allocate_temp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.h | 181 gtl::ArraySlice<int> candidate_input_indices, DataType type,
|
| D | cuda_solvers.cc | 284 gtl::ArraySlice<int> candidate_input_indices, DataType type, in forward_input_or_allocate_scoped_tensor() argument
287 candidate_input_indices, type, shape, out_temp); in forward_input_or_allocate_scoped_tensor()
|