Searched refs:mutable_result (Results 1 – 16 of 16) sorted by relevance
200 *mutable_result() = Shape(program_shape_proto.result()); in ProgramShape()211 *proto.mutable_result() = result().ToProto(); in ToProto()
172 LayoutUtil::SetToDefaultLayout(program_shape->mutable_result()); in SetToDefaultLayout()272 LayoutUtil::ClearLayout(program_shape->mutable_result()); in ClearLayout()
155 *program_shape.mutable_result() = ShapeUtil::MakeShape(F32, {7}); in TEST_F()
303 Shape* mutable_result() { return &result_; } in mutable_result() function
248 *program_shape.mutable_result() = std::move(result); in MakeProgramShape()
654 *shapes1->mutable_result() = in TEST()665 *shapes2->mutable_result() = in TEST()891 *shapes->mutable_result() = in TEST()961 *shapes->mutable_result() = in TEST()1083 *shapes->mutable_result() = in TEST()1141 *shapes->mutable_result() = dyn_shape.ToProto(); in TEST()1198 *shapes->mutable_result() = dyn_shape.ToProto(); in TEST()1250 *shapes->mutable_result() = in TEST()1316 *shapes->mutable_result() = dyn_shape.ToProto(); in TEST()1369 *shapes->mutable_result() = dyn_shape.ToProto(); in TEST()[all …]
79 *program_shape.mutable_result() = result_layout_.shape(); in ComputeProgramShape()
843 *program_shape.mutable_result() = root_instruction_->shape(); in ComputeProgramShape()
94 *module->mutable_result() = literal.ToProto(); in RecordResult()
214 *response->mutable_result() = GetHandshakerResult(); in ProcessNext()
230 *snapshot->mutable_result() = literal->ToProto(); in DumpOutputsAndSaveSnapshot()
283 *program_shape.mutable_result() = Shape(root_proto->shape()); in GetProgramShape()3417 *program_shape->mutable_result() = in BuildDynamicInferenceGraph()3778 *program_shape->mutable_result() = root->shape(); in BuildConstantSubGraph()
264 *program_shape.mutable_result() = result; in BuildXlaCompilerSubmodule()
82 *program_shape.mutable_result() = entry->root_instruction()->shape(); in GetProgramShapeWithLayout()
779 *hlo_snapshot->mutable_result() = literal->ToProto(); in DoWork()
3708 old_entry_layout.result_shape(), new_program_shape.mutable_result())); in Run()