Home
last modified time | relevance | path

Searched refs:parameter_layout (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_layout.cc41 for (auto& parameter_layout : parameter_layouts_) { in SetToDefaultLayout() local
42 parameter_layout.SetToDefaultLayout(); in SetToDefaultLayout()
48 for (auto& parameter_layout : parameter_layouts_) { in SetToDefaultLayoutIfEmpty() local
49 if (!parameter_layout.LayoutIsSet()) { in SetToDefaultLayoutIfEmpty()
50 parameter_layout.SetToDefaultLayout(); in SetToDefaultLayoutIfEmpty()
95 for (const auto& parameter_layout : parameter_layouts_) { in Hash() local
97 hash_value, ShapeUtil::Hash(parameter_layout.shape())); in Hash()
Dlayout_assignment.cc473 const ShapeLayout& parameter_layout = in AddMandatoryConstraints() local
474 computation_layout->parameter_layout( in AddMandatoryConstraints()
479 parameter_layout.shape(), instruction)); in AddMandatoryConstraints()
562 called_computation_layout.parameter_layout(i).shape(), instruction, in AddMandatoryConstraints()
586 if (body_layout.result_layout() != body_layout.parameter_layout(0)) { in AddMandatoryConstraints()
592 if (condition_layout.parameter_layout(0) != in AddMandatoryConstraints()
593 body_layout.parameter_layout(0)) { in AddMandatoryConstraints()
596 << " shape=" << body_layout.parameter_layout(0).ToString(); in AddMandatoryConstraints()
598 body_layout.parameter_layout(0); in AddMandatoryConstraints()
679 TF_RET_CHECK(computation_layout.parameter_layout(i).MatchesLayoutInShape( in CheckCallLayout()
[all …]
Dcomputation_layout.h51 const ShapeLayout& parameter_layout(int64 param_no) const { in parameter_layout() function
Dhlo_module.cc345 module_config.entry_computation_layout().parameter_layout(i).shape(); in CreateFromProto()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc132 if (!computation_layout.parameter_layout(i).MatchesLayoutInShape( in RunHelper()
141 computation_layout.parameter_layout(i).shape()), in RunHelper()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc474 op::ShapeWithLayout(computation_layout.parameter_layout(0).shape()), in TEST_F()
476 computation_layout.parameter_layout(1).shape())))); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc691 .parameter_layout(i) in MakeFakeArguments()
695 .parameter_layout(i) in MakeFakeArguments()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.cc444 layout.parameter_layout(i).shape(), in ToC()