Searched refs:GetOperandShapes (Results 1 – 2 of 2) sorted by relevance
111 StatusOr<std::vector<Shape>> XlaBuilder::GetOperandShapes( in GetOperandShapes() function in xla::XlaBuilder616 TF_ASSIGN_OR_RETURN(const auto& operand_shapes, GetOperandShapes(operands)); in Call()789 GetOperandShapes(start_indices)); in DynamicSlice()836 GetOperandShapes(start_indices)); in DynamicUpdateSlice()858 TF_ASSIGN_OR_RETURN(const auto& operand_shapes, GetOperandShapes(operands)); in ConcatInDim()980 TF_ASSIGN_OR_RETURN(const auto& operand_shapes, GetOperandShapes(elements)); in Tuple()1691 GetOperandShapes(operands)); in Sort()1748 TF_ASSIGN_OR_RETURN(const auto& operand_shapes, GetOperandShapes(operands)); in Map()1963 GetOperandShapes(all_operands)); in Reduce()2190 TF_ASSIGN_OR_RETURN(auto slice_shapes, this->GetOperandShapes(slices)); in AllToAll()
629 StatusOr<std::vector<Shape>> GetOperandShapes(