Searched defs:candidate_input_indices (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| H A D | tf_framework_c_interface.cc | 65 int32_t* candidate_input_indices) { in _mlir_ciface_tf_alloc()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel.cc | 580 gtl::ArraySlice<int> candidate_input_indices, DataType type, in forward_input_or_allocate_temp()
596 gtl::ArraySlice<int> candidate_input_indices, int output_index, in forward_input_or_allocate_output()
|
| H A D | op_kernel.h | 901 gtl::ArraySlice<int> candidate_input_indices, DataType type, in forward_input_or_allocate_temp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | rocm_solvers.cc | 721 gtl::ArraySlice<int> candidate_input_indices, DataType type, in forward_input_or_allocate_scoped_tensor()
|
| H A D | cuda_solvers.cc | 304 gtl::ArraySlice<int> candidate_input_indices, DataType type, in forward_input_or_allocate_scoped_tensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | kernels.cc | 707 TF_OpKernelContext* context, const int* candidate_input_indices, in TF_ForwardInputOrAllocateOutput()
|
| H A D | kernels_test.cc | 1337 int candidate_input_indices[1] = {0}; in TEST_F() local
|