Home
last modified time | relevance | path

Searched defs:execution_input (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc214 xla::ExecutionInput* execution_input = &(*execution_inputs)[i]; in UpdateDynamicInputs() local
444 const Tensor& execution_input = context->input(0); in DoWork() local
Dtpu_execute_op.cc295 const Tensor& execution_input = context->input(0); in DoWork() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer.cc153 ExecutionInput* execution_input, in AddToInputAsDonated()
Dpjrt_stream_executor_client.cc566 ExecutionInput* execution_input, in AddToInput()
1592 ExecutionInput execution_input; member
1664 ExecutionInput execution_input(tupled_parameter_shape); in MakeTupleHelper() local
1832 ExecutionInput& execution_input = execution_inputs.back(); in MakeExecutionInputsAndWaitForEvents() local
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc213 static void PopulateExecutionInputBuffer(xla::ExecutionInput& execution_input, in PopulateExecutionInputBuffer()
282 xla::ExecutionInput& execution_input = arguments.back(); in PopulateInputs() local