Searched refs:mutable_result (Results 1 – 15 of 15) sorted by relevance
190 *mutable_result() = Shape(program_shape_proto.result()); in ProgramShape()201 *proto.mutable_result() = result().ToProto(); in ToProto()
170 LayoutUtil::SetToDefaultLayout(program_shape->mutable_result()); in SetToDefaultLayout()270 LayoutUtil::ClearLayout(program_shape->mutable_result()); in ClearLayout()
152 *program_shape.mutable_result() = ShapeUtil::MakeShape(F32, {7}); in TEST_F()
252 Shape* mutable_result() { return &result_; } in mutable_result() function
154 *program_shape.mutable_result() = std::move(result); in MakeProgramShape()
454 *shapes1->mutable_result() = in TEST()465 *shapes2->mutable_result() = in TEST()702 *shapes->mutable_result() = in TEST()755 *shapes->mutable_result() = in TEST()822 *shapes->mutable_result() = result_shape.ToProto(); in TEST()876 *shapes->mutable_result() = in TEST()918 *shapes->mutable_result() = xla::ShapeUtil::MakeShape(xla::F32, {}).ToProto(); in TEST()960 *shapes->mutable_result() = in TEST()1010 *shapes->mutable_result() = in TEST()1068 *shapes->mutable_result() = in TEST()[all …]
65 *program_shape.mutable_result() = result_layout_.shape(); in ComputeProgramShape()
714 *program_shape.mutable_result() = root_instruction_->shape(); in ComputeProgramShape()
93 *module->mutable_result() = literal.ToProto(); in RecordResult()
214 *response->mutable_result() = GetHandshakerResult(); in ProcessNext()
208 *hlo_snapshot->mutable_result() = literal.ToProto(); in RecordResult()
161 *program_shape.mutable_result() = Shape(root_proto->shape()); in GetProgramShape()2546 *program_shape->mutable_result() = root->shape(); in BuildConstantSubGraph()
158 *program_shape.mutable_result() = result_shape; in CompileForXrt()
406 return std::move(*shape.mutable_result()); in GetReturnValueShape()
82 *program_shape.mutable_result() = entry->root_instruction()->shape(); in GetProgramShapeWithLayout()