Home
last modified time | relevance | path

Searched defs:param_layout (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_config.cc55 for (const ShapeLayout& param_layout : in compilation_cache_key() local
H A Dcomputation_layout.cc67 for (auto& param_layout : parameter_layouts_) { in ToString() local
H A Dlayout_assignment.cc2091 auto param_layout = InferArrayLayout(operand, index); in CalculateComputationLayout() local
2368 ShapeLayout* param_layout = computation_layout->mutable_parameter_layout(i); in PropagateComputationLayouts() local
H A Dhlo_parser_test.cc2889 auto param_layout = program_layout.parameter_layout(0).layout(); local