Searched refs:branch_computation_shapes (Results 1 – 2 of 2) sorted by relevance
2349 absl::InlinedVector<ProgramShape, 2> branch_computation_shapes; in ParseInstructionRhs() local2350 branch_computation_shapes.reserve(branch_computations->size()); in ParseInstructionRhs()2352 branch_computation_shapes.push_back( in ParseInstructionRhs()2361 operands[0]->shape(), branch_computation_shapes, in ParseInstructionRhs()
2415 std::vector<ProgramShape> branch_computation_shapes( in ConditionalImpl() local2420 TF_ASSIGN_OR_RETURN(branch_computation_shapes[j], in ConditionalImpl()2425 *branch_index_shape, branch_computation_shapes, in ConditionalImpl()