Searched refs:parameter_layout (Results 1 – 10 of 10) sorted by relevance
41 for (auto& parameter_layout : parameter_layouts_) { in SetToDefaultLayout() local42 parameter_layout.SetToDefaultLayout(); in SetToDefaultLayout()48 for (auto& parameter_layout : parameter_layouts_) { in SetToDefaultLayoutIfEmpty() local49 if (!parameter_layout.LayoutIsSet()) { in SetToDefaultLayoutIfEmpty()50 parameter_layout.SetToDefaultLayout(); in SetToDefaultLayoutIfEmpty()
51 const ShapeLayout& parameter_layout(int64_t param_no) const { in parameter_layout() function98 for (const auto& parameter_layout : computation_layout.parameter_layouts_) { in AbslHashValue() local99 h = H::combine(std::move(h), parameter_layout.shape()); in AbslHashValue()
608 const ShapeLayout& parameter_layout = in AddMandatoryConstraints() local609 constraints->computation_layout().parameter_layout( in AddMandatoryConstraints()614 SetInstructionLayout(parameter_layout.shape(), instruction)); in AddMandatoryConstraints()617 instruction, parameter_layout.shape(), this)); in AddMandatoryConstraints()677 called_computation_layout.parameter_layout(i).shape(), instruction, in AddMandatoryConstraints()709 if (body_layout.result_layout() != body_layout.parameter_layout(0)) { in AddMandatoryConstraints()719 if (condition_layout.parameter_layout(0) != in AddMandatoryConstraints()720 body_layout.parameter_layout(0)) { in AddMandatoryConstraints()723 << " shape=" << body_layout.parameter_layout(0).ToString(); in AddMandatoryConstraints()725 body_layout.parameter_layout(0); in AddMandatoryConstraints()[all …]
170 module->entry_computation_layout().parameter_layout(i).shape(); in UpdateEntryComputationLayout()
404 module_config.entry_computation_layout().parameter_layout(i).shape(); in CreateFromProto()
459 .parameter_layout(i) in RunAndCompareTwoModules()463 .parameter_layout(i) in RunAndCompareTwoModules()471 .parameter_layout(i) in RunAndCompareTwoModules()475 .parameter_layout(i) in RunAndCompareTwoModules()
748 .parameter_layout(i) in MakeFakeArguments()752 .parameter_layout(i) in MakeFakeArguments()
134 if (!computation_layout.parameter_layout(i).MatchesLayoutInShape( in RunHelper()144 computation_layout.parameter_layout(i).shape()), in RunHelper()
472 op::ShapeWithLayout(computation_layout.parameter_layout(0).shape()), in TEST_F()474 computation_layout.parameter_layout(1).shape())))); in TEST_F()
489 layout.parameter_layout(i).shape(), in ToC()