Searched refs:computed_computation_layout (Results 1 – 1 of 1) sorted by relevance
1891 ComputationLayout computed_computation_layout( in PropagateComputationLayouts() local1894 for (int64 i = 0; i < computed_computation_layout.parameter_count(); ++i) { in PropagateComputationLayouts()1899 << computed_computation_layout.parameter_layout(i).ToString(); in PropagateComputationLayouts()1900 *param_layout = computed_computation_layout.parameter_layout(i); in PropagateComputationLayouts()1902 TF_RET_CHECK(computed_computation_layout.parameter_layout(i) == in PropagateComputationLayouts()1909 << ": " << computed_computation_layout.result_layout().ToString(); in PropagateComputationLayouts()1910 *result_layout = computed_computation_layout.result_layout(); in PropagateComputationLayouts()1912 TF_RET_CHECK(computed_computation_layout.result_layout() == *result_layout); in PropagateComputationLayouts()