Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/kernels/
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
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.cc74 StatusOr<std::vector<xla::ExecutionInput>> execution_inputs = in Run() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc84 std::vector<ExecutionInput> execution_inputs; in RunMlirModule() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner.cc133 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.cc1848 std::vector<ExecutionInput> execution_inputs; in MakeExecutionInputsAndWaitForEvents() local