Searched defs:input_handles (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/aoti_runtime/ |
| H A D | implementation.cpp | 63 AtenTensorHandle* input_handles, in convert_handles_to_inputs_helper()
72 AtenTensorHandle* input_handles, in convert_handles_to_inputs()
|
| H A D | interface.cpp | 96 AtenTensorHandle* input_handles, // array of input AtenTensorHandle; handles in AOTInductorModelContainerRun()
306 AtenTensorHandle* input_handles, in AOTInductorModelRun()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | sharding_util_ops.cc | 316 std::vector<xla::XlaOp> input_handles; in CompileInternal() local
398 std::vector<xla::XlaOp>& input_handles, in GetInputsAndOutputShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/ |
| H A D | graph_jni.cc | 213 jlongArray input_handles = env->NewLongArray(ninputs); in buildSubgraph() local
257 JNIEnv* env, jclass clazz, jlong handle, jlongArray input_handles, in Java_org_tensorflow_Graph_whileLoop()
|
| H A D | eager_operation_builder_jni.cc | 141 JNIEnv* env, jclass clazz, jlong op_handle, jlongArray input_handles) { in Java_org_tensorflow_EagerOperationBuilder_addInputList()
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runner/ |
| H A D | model_container_runner.cpp | 101 auto input_handles = in run() local
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/detail/ |
| H A D | CrossLayerMemoryManagerHelpers.cpp | 56 …std::vector<std::pair<ITensorHandle *, IMemoryGroup *>> input_handles = {}; /**< Input handles to… member
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runtime/ |
| H A D | model_container.h | 73 input_handles, // array of input AtenTensorHandle; handles in run()
|
| H A D | model.h | 133 input_handles, // array of input AtenTensorHandle; handles in run()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| H A D | host_compute_ops.cc | 145 std::vector<xla::XlaOp> input_handles; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | list_ops.cc | 129 shape_inference::ShapeHandle input_handles; in __anon5e0c1ea50402() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ |
| H A D | tfe_wrapper.cc | 440 std::vector<const TensorHandle*> input_handles; in InputTFE_OutputTensorHandles() local
|