Searched refs:InputCoords (Results 1 – 4 of 4) sorted by relevance
53 struct InputCoords { struct54 explicit InputCoords(int64 handle) : handle(handle) {} in InputCoords() function55 InputCoords(int64 handle, xla::ShapeIndex index) in InputCoords() argument71 xla::StatusOr<std::vector<InputCoords>> GetComputationInputs( argument78 const std::vector<InputCoords>& input_coords,
202 xla::StatusOr<std::vector<InputCoords>> GetComputationInputs( in GetComputationInputs()207 std::vector<InputCoords> input_coords; in GetComputationInputs()256 const std::vector<InputCoords>& input_coords, in GetInputTupleAllocations()
90 xla::Backend* backend, const std::vector<InputCoords>& input_coords, in GetInputTuples()398 const std::vector<InputCoords>& input_coords, bool release_inputs, in ExecuteComputation()475 TF_ASSIGN_OR_RETURN(std::vector<InputCoords> input_coords, in DoWork()
337 TF_ASSIGN_OR_RETURN(std::vector<InputCoords> input_coords, in DoWork()