Searched defs:execution_input (Results 1 – 5 of 5) sorted by relevance
214 xla::ExecutionInput* execution_input = &(*execution_inputs)[i]; in UpdateDynamicInputs() local442 const Tensor& execution_input = context->input(0); in DoWork() local
293 const Tensor& execution_input = context->input(0); in DoWork() local
151 ExecutionInput* execution_input, in AddToInputAsDonated()
566 ExecutionInput* execution_input, in AddToInput()1473 ExecutionInput execution_input; member1516 ExecutionInput execution_input(on_device_shape); in MakeTupleHelper() local1658 ExecutionInput& execution_input = execution_inputs.back(); in MakeExecutionInputsAndWaitForEvents() local
204 static void PopulateExecutionInputBuffer(xla::ExecutionInput& execution_input, in PopulateExecutionInputBuffer()269 xla::ExecutionInput& execution_input = arguments.back(); in PopulateInputs() local