Searched refs:computed_computation_layout (Results 1 – 1 of 1) sorted by relevance
2048 ComputationLayout computed_computation_layout( in PropagateComputationLayouts() local2051 for (int64 i = 0; i < computed_computation_layout.parameter_count(); ++i) { in PropagateComputationLayouts()2065 computed_computation_layout.parameter_shape(i), shape_index); in PropagateComputationLayouts()2070 computed_computation_layout.ToString(), in PropagateComputationLayouts()2078 << computed_computation_layout.parameter_layout(i).ToString(); in PropagateComputationLayouts()2079 *param_layout = computed_computation_layout.parameter_layout(i); in PropagateComputationLayouts()2085 << ": " << computed_computation_layout.result_layout().ToString(); in PropagateComputationLayouts()2086 *result_layout = computed_computation_layout.result_layout(); in PropagateComputationLayouts()2090 computed_computation_layout.result_layout().shape(), in PropagateComputationLayouts()