Searched refs:ValidateShape (Results 1 – 13 of 13) sorted by relevance
67 TF_CHECK_OK(ShapeUtil::ValidateShape(shape_)); in ResetLayout()75 TF_CHECK_OK(ShapeUtil::ValidateShape(shape_)); in ResetLayout()
110 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(shape)); in MakeShapeWithLayoutInternal()211 TF_DCHECK_OK(ShapeUtil::ValidateShape(shape)); in MakeShapeWithSparseLayout()234 return ValidateShape(*shape); in PopulateShape()272 TF_DCHECK_OK(ValidateShape(*shape)); in AppendMajorDimension()572 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOf()593 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOfTupleIndexTable()600 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOfElements()615 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOfSparseIndices()756 /* static */ Status ShapeUtil::ValidateShape(const Shape& shape) { in ValidateShape() function in xla::ShapeUtil
426 static Status ValidateShape(const Shape& shape);
69 Status ValidateShape(const Tensor* tensor) in ValidateShape() function125 OP_REQUIRES_OK_BOOLEAN(ctx, this->ValidateShape(*tensor)); in GetAndValidateTensorInputForApplyGrad()
84 Status ValidateShape( in ValidateShape() function399 OP_REQUIRES_OK_BOOLEAN(ctx, this->ValidateShape(*tensor, has_known_shape)); in GetAndValidateTensorInputForApplyGrad()
196 Status ValidateShape(Node* node);586 Status GraphConstructor::ValidateShape(Node* node) { in ValidateShape() function in tensorflow::__anone0324be20111::GraphConstructor1045 TF_RETURN_IF_ERROR(ValidateShape(node)); in Convert()
316 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(on_host_shape)); in AllocateScopedShapedBuffer()
1087 TF_DCHECK_OK(ShapeUtil::ValidateShape(program_shape.result())); in ComputeConstantGraph()
1017 TF_CHECK_OK(ShapeUtil::ValidateShape(lhs_shape)); in HandleConvolution()1018 TF_CHECK_OK(ShapeUtil::ValidateShape(rhs_shape)); in HandleConvolution()
1599 return ShapeUtil::ValidateShape(hlo->shape()); in Preprocess()
885 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(instruction->shape())); in CheckLayouts()
103 TF_CHECK_OK(ShapeUtil::ValidateShape(shape)); in IrArray()
253 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(shape)); in DecodeSelfDescribingShapeConstant()