Searched refs:computed_computation_layout (Results 1 – 1 of 1) sorted by relevance
1087 ComputationLayout computed_computation_layout( in CopyOperandIfLayoutsDiffer() local1092 ->ResetComputationLayout(computed_computation_layout, in CopyOperandIfLayoutsDiffer()2364 ComputationLayout computed_computation_layout( in PropagateComputationLayouts() local2367 for (int64_t i = 0; i < computed_computation_layout.parameter_count(); ++i) { in PropagateComputationLayouts()2381 computed_computation_layout.parameter_shape(i), shape_index); in PropagateComputationLayouts()2386 computed_computation_layout.ToString(), in PropagateComputationLayouts()2394 << computed_computation_layout.parameter_layout(i).ToString(); in PropagateComputationLayouts()2395 *param_layout = computed_computation_layout.parameter_layout(i); in PropagateComputationLayouts()2401 << ": " << computed_computation_layout.result_layout().ToString(); in PropagateComputationLayouts()2402 *result_layout = computed_computation_layout.result_layout(); in PropagateComputationLayouts()[all …]