Home
last modified time | relevance | path

Searched defs:execution_inputs (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/kernels/
H A Dxla_ops.cc173 std::vector<xla::ExecutionInput> execution_inputs, in RunExecutable()
348 StatusOr<std::vector<xla::ExecutionInput>> execution_inputs = in ComputeAsync() local
591 StatusOr<std::vector<xla::ExecutionInput>> execution_inputs; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_compile_on_demand_op.cc74 StatusOr<std::vector<xla::ExecutionInput>> execution_inputs = in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dmlir_gpu_test_base.cc84 std::vector<ExecutionInput> execution_inputs; in RunMlirModule() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/
H A Dxrt_execute_op.cc203 std::vector<xla::ExecutionInput>* execution_inputs, in UpdateDynamicInputs()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_runner.cc133 std::vector<ExecutionInput> execution_inputs; in ExecutionInputsFromScopedShapedBuffers() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_stream_executor_client.cc1848 std::vector<ExecutionInput> execution_inputs; in MakeExecutionInputsAndWaitForEvents() local