Home
last modified time | relevance | path

Searched refs:ValidateShape (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_layout.cc67 TF_CHECK_OK(ShapeUtil::ValidateShape(shape_)); in ResetLayout()
75 TF_CHECK_OK(ShapeUtil::ValidateShape(shape_)); in ResetLayout()
Dshape_util.cc110 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
Dshape_util.h426 static Status ValidateShape(const Shape& shape);
/external/tensorflow/tensorflow/core/kernels/
Dconditional_accumulator.h69 Status ValidateShape(const Tensor* tensor) in ValidateShape() function
125 OP_REQUIRES_OK_BOOLEAN(ctx, this->ValidateShape(*tensor)); in GetAndValidateTensorInputForApplyGrad()
Dsparse_conditional_accumulator.h84 Status ValidateShape( in ValidateShape() function
399 OP_REQUIRES_OK_BOOLEAN(ctx, this->ValidateShape(*tensor, has_known_shape)); in GetAndValidateTensorInputForApplyGrad()
/external/tensorflow/tensorflow/core/graph/
Dgraph_constructor.cc196 Status ValidateShape(Node* node);
586 Status GraphConstructor::ValidateShape(Node* node) { in ValidateShape() function in tensorflow::__anone0324be20111::GraphConstructor
1045 TF_RETURN_IF_ERROR(ValidateShape(node)); in Convert()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.cc316 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(on_host_shape)); in AllocateScopedShapedBuffer()
Dservice.cc1087 TF_DCHECK_OK(ShapeUtil::ValidateShape(program_shape.result())); in ComputeConstantGraph()
Dhlo_evaluator_typed_visitor.h1017 TF_CHECK_OK(ShapeUtil::ValidateShape(lhs_shape)); in HandleConvolution()
1018 TF_CHECK_OK(ShapeUtil::ValidateShape(rhs_shape)); in HandleConvolution()
Dhlo_evaluator.cc1599 return ShapeUtil::ValidateShape(hlo->shape()); in Preprocess()
Dlayout_assignment.cc885 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(instruction->shape())); in CheckLayouts()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc103 TF_CHECK_OK(ShapeUtil::ValidateShape(shape)); in IrArray()
Dllvm_util.cc253 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(shape)); in DecodeSelfDescribingShapeConstant()