Searched defs:program_shape (Results 1 – 18 of 18) sorted by relevance
33 HloModuleConfig::HloModuleConfig(const ProgramShape& program_shape) in HloModuleConfig()37 const ProgramShape& program_shape) { in SetDefaultComputationLayout()
26 ComputationLayout::ComputationLayout(const ProgramShape& program_shape) in ComputationLayout()
265 const ProgramShape& program_shape, in CreateModuleConfig()325 const ProgramShape& program_shape, in CreateModuleConfig()1235 ProgramShape program_shape; in ComputeConstant() local
260 ProgramShape program_shape; in ProgramShapeFromProto() local
527 ProgramShape program_shape; in ComputeProgramShape() local
1621 auto program_shape = MakeUnique<ProgramShape>(); in ComputeProgramShape() local
36 xla::ProgramShape program_shape; // Static shape of args and results. member
539 std::unique_ptr<xla::ProgramShape> program_shape; in GenerateMetadata() local
135 /* static */ void LayoutUtil::SetToDefaultLayout(ProgramShape* program_shape) { in SetToDefaultLayout()237 /* static */ void LayoutUtil::ClearLayout(ProgramShape* program_shape) { in ClearLayout()322 /* static */ bool LayoutUtil::HasLayout(const ProgramShape& program_shape) { in HasLayout()
174 ProgramShape program_shape; in MakeProgramShape() local484 /* static */ string ShapeUtil::HumanString(const ProgramShape& program_shape) { in HumanString()
464 ProgramShape program_shape = 1; field
101 auto program_shape = in MakeFakeArgumentsOrDie() local
55 std::unique_ptr<ProgramShape> program_shape = in RealMain() local
80 std::unique_ptr<ProgramShape> program_shape = in RealMain() local
119 const xla::ProgramShape* program_shape = function.ProgramShape(); in TEST() local
40 const xla::ProgramShape& program_shape) { in ComputeArgSizes()
78 const xla::ProgramShape* program_shape = nullptr; member
76 ProgramShape program_shape; in GetProgramShapeWithLayout() local128 const auto& program_shape = GetProgramShapeWithLayout(test_module); in MakeReferenceModule() local