Searched refs:mutable_result (Results 1 – 14 of 14) sorted by relevance
193 *mutable_result() = Shape(program_shape_proto.result()); in ProgramShape()204 *proto.mutable_result() = result().ToProto(); in ToProto()
166 LayoutUtil::SetToDefaultLayout(program_shape->mutable_result()); in SetToDefaultLayout()266 LayoutUtil::ClearLayout(program_shape->mutable_result()); in ClearLayout()
154 *program_shape.mutable_result() = ShapeUtil::MakeShape(F32, {7}); in TEST_F()
287 Shape* mutable_result() { return &result_; } in mutable_result() function
170 *program_shape.mutable_result() = std::move(result); in MakeProgramShape()
569 *shapes1->mutable_result() = in TEST()580 *shapes2->mutable_result() = in TEST()806 *shapes->mutable_result() = in TEST()876 *shapes->mutable_result() = in TEST()998 *shapes->mutable_result() = in TEST()1051 *shapes->mutable_result() = in TEST()1118 *shapes->mutable_result() = result_shape.ToProto(); in TEST()1173 *shapes->mutable_result() = in TEST()1215 *shapes->mutable_result() = xla::ShapeUtil::MakeShape(xla::F32, {}).ToProto(); in TEST()1257 *shapes->mutable_result() = in TEST()[all …]
66 *program_shape.mutable_result() = result_layout_.shape(); in ComputeProgramShape()
823 *program_shape.mutable_result() = root_instruction_->shape(); in ComputeProgramShape()
93 *module->mutable_result() = literal.ToProto(); in RecordResult()
214 *response->mutable_result() = GetHandshakerResult(); in ProcessNext()
221 *snapshot->mutable_result() = literal->ToProto(); in DumpOutputsAndSaveSnapshot()
208 *program_shape.mutable_result() = Shape(root_proto->shape()); in GetProgramShape()2589 *program_shape->mutable_result() = root->shape(); in BuildConstantSubGraph()
82 *program_shape.mutable_result() = entry->root_instruction()->shape(); in GetProgramShapeWithLayout()
443 *program_shape.mutable_result() = result; in PYBIND11_MODULE()