Searched defs:candidate_input_indices (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_framework_c_interface.cc | 43 int32_t* candidate_input_indices) { in _mlir_ciface_tf_alloc()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | rocm_solvers.cc | 197 gtl::ArraySlice<int> candidate_input_indices, DataType type, in forward_input_or_allocate_scoped_tensor()
|
| D | cuda_solvers.cc | 305 gtl::ArraySlice<int> candidate_input_indices, DataType type, in forward_input_or_allocate_scoped_tensor()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 920 gtl::ArraySlice<int> candidate_input_indices, DataType type, in forward_input_or_allocate_temp()
1637 gtl::ArraySlice<int> candidate_input_indices, int output_index, in forward_input_or_allocate_output()
|
| D | op_kernel.cc | 567 gtl::ArraySlice<int> candidate_input_indices, DataType type, in forward_input_or_allocate_temp()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels.cc | 484 TF_OpKernelContext* context, int* candidate_input_indices, in TF_ForwardInputOrAllocateOutput()
|
| D | kernels_test.cc | 929 int candidate_input_indices[1] = {0}; in TEST_F() local
|