Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_layout.cc72 TF_CHECK_OK(ShapeUtil::ValidateShape(shape_)); in ResetLayout()
80 TF_CHECK_OK(ShapeUtil::ValidateShape(shape_)); in ResetLayout()
Dshape_util.cc115 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(shape)); in MakeShapeWithLayoutInternal()
261 return ValidateShape(*shape); in PopulateShape()
319 TF_DCHECK_OK(ValidateShape(*shape)); in AppendMajorDimension()
625 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOf()
643 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOfTupleIndexTable()
650 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOfElements()
756 /* static */ Status ShapeUtil::ValidateShape(const Shape& shape) { in ValidateShape() function in xla::ShapeUtil
Dshape_util.h439 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/compiler/xla/service/
Dtransfer_manager.cc321 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(on_host_shape)); in AllocateScopedShapedBuffer()
Dhlo_evaluator.cc1259 TF_CHECK_OK(ShapeUtil::ValidateShape(input_shape)); in CheckParameters()
1278 TF_CHECK_OK(ShapeUtil::ValidateShape(output_shape)); in CheckParameters()
2501 return ShapeUtil::ValidateShape(hlo->shape()); in Preprocess()
Dservice.cc1108 TF_DCHECK_OK(ShapeUtil::ValidateShape(program_shape.result())); in ComputeConstantGraph()
Dhlo_evaluator_typed_visitor.h1045 TF_CHECK_OK(ShapeUtil::ValidateShape(lhs_shape));
1046 TF_CHECK_OK(ShapeUtil::ValidateShape(rhs_shape));
Dlayout_assignment.cc937 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(instruction->shape())); in CheckLayouts()
/external/tensorflow/tensorflow/core/graph/
Dgraph_constructor.cc229 Status ValidateShape(Node* node);
750 Status GraphConstructor::ValidateShape(Node* node) { in ValidateShape() function in tensorflow::__anon29e8b4a40111::GraphConstructor
1289 TF_RETURN_IF_ERROR(ValidateShape(node)); in Convert()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc133 auto status = xla::ShapeUtil::ValidateShape(shape); in DecodeSelfDescribingShapeConstant()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc109 TF_CHECK_OK(ShapeUtil::ValidateShape(shape)); in IrArray()