Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.cc71 xla::StatusOr<std::vector<xla::ExecutionInput>> execution_inputs = in Run() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc90 std::vector<ExecutionInput> execution_inputs; in RunMlirModule() local
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc263 xla::StatusOr<std::vector<xla::ExecutionInput>> execution_inputs = in Compute() local
485 xla::StatusOr<std::vector<xla::ExecutionInput>> execution_inputs; in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner.cc125 std::vector<ExecutionInput> execution_inputs; in ExecutionInputsFromScopedShapedBuffers() local
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc203 std::vector<xla::ExecutionInput>* execution_inputs, in UpdateDynamicInputs()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1640 std::vector<ExecutionInput> execution_inputs; in MakeExecutionInputsAndWaitForEvents() local