Searched refs:param_layout (Results 1 – 3 of 3) sorted by relevance
55 for (const ShapeLayout& param_layout : in compilation_cache_key() local57 params.push_back(param_layout.shape().DebugString()); in compilation_cache_key()
66 for (auto& param_layout : parameter_layouts_) { in ToString() local67 params.push_back(param_layout.ToString()); in ToString()
2052 ShapeLayout* param_layout = computation_layout->mutable_parameter_layout(i); in PropagateComputationLayouts() local2055 param_layout->shape(), in PropagateComputationLayouts()2057 if (!ShapeUtil::IsLeafIndex(param_layout->shape(), shape_index)) { in PropagateComputationLayouts()2079 *param_layout = computed_computation_layout.parameter_layout(i); in PropagateComputationLayouts()