Searched refs:branch_computation_shapes (Results 1 – 2 of 2) sorted by relevance
2212 absl::InlinedVector<ProgramShape, 2> branch_computation_shapes; in ParseInstructionRhs() local2213 branch_computation_shapes.reserve(branch_computations->size()); in ParseInstructionRhs()2215 branch_computation_shapes.push_back( in ParseInstructionRhs()2224 operands[0]->shape(), branch_computation_shapes, in ParseInstructionRhs()
2387 std::vector<ProgramShape> branch_computation_shapes( in ConditionalImpl() local2392 TF_ASSIGN_OR_RETURN(branch_computation_shapes[j], in ConditionalImpl()2397 *branch_index_shape, branch_computation_shapes, in ConditionalImpl()