Searched refs:input_tensor_handles (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/ |
| H A D | session_jni.cc | 134 jlongArray input_tensor_handles, jlongArray input_op_handles, in Java_org_tensorflow_Session_run() argument
141 const jint ninputs = env->GetArrayLength(input_tensor_handles); in Java_org_tensorflow_Session_run()
153 resolveHandles(env, "input Tensors", input_tensor_handles, input_values.get(), in Java_org_tensorflow_Session_run()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A D | runtime_fallback_kernels.cc | 549 llvm::ArrayRef<TensorHandle*> input_tensor_handles, in CallEagerExecute() argument
560 for (TensorHandle* input_tensor : input_tensor_handles) { in CallEagerExecute()
599 llvm::SmallVector<TensorHandle*, 4> input_tensor_handles; in RuntimeFallbackExecute() local
600 input_tensor_handles.reserve(arguments.size()); in RuntimeFallbackExecute()
602 input_tensor_handles.push_back( in RuntimeFallbackExecute()
613 input_tensor_handles, attrs, result_tensor_handles); in RuntimeFallbackExecute()
626 exec_ctx, eager_ctx, op_name, device_name, input_tensor_handles, in RuntimeFallbackExecute()
|
| H A D | runtime_fallback_kernels.h | 46 llvm::ArrayRef<TensorHandle*> input_tensor_handles,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ |
| H A D | tfe_wrapper.cc | 122 TFE_InputTensorHandles input_tensor_handles; in InputTFE_InputTensorHandles() local
128 input_tensor_handles.resize(len); in InputTFE_InputTensorHandles()
135 (input_tensor_handles)[i] = EagerTensor_Handle(elem); in InputTFE_InputTensorHandles()
150 (input_tensor_handles)[i] = EagerTensor_Handle(value_attr.get()); in InputTFE_InputTensorHandles()
224 return input_tensor_handles; in InputTFE_InputTensorHandles()
339 TFE_InputTensorHandles input_tensor_handles = in InputTFE_OutputTensorHandles() local
344 TFE_Py_ExecuteCancelable(ctx, device_name, op_name, &input_tensor_handles, in InputTFE_OutputTensorHandles()
|