Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconditional_accumulator.h67 Status ValidateShape(const Tensor* tensor) in ValidateShape() function
123 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/
Dshape_util.cc139 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(shape)); in MakeShapeWithLayoutInternal()
212 TF_DCHECK_OK(ShapeUtil::ValidateShape(shape)); in MakeShapeWithSparseLayout()
235 TF_DCHECK_OK(ValidateShape(*shape)); in PopulateShape()
266 TF_DCHECK_OK(ValidateShape(*shape)); in AppendMajorDimension()
591 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(result)); in ParseShapeStringInternal()
702 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOf()
716 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOfTupleIndexTable()
723 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOfElements()
748 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOfSparseIndices()
798 return ValidateShape(shape); in ValidateShapeWithOptionalLayout()
[all …]
Dshape_layout.cc67 TF_CHECK_OK(ShapeUtil::ValidateShape(shape_)); in ResetLayout()
Dshape_util.h352 static Status ValidateShape(const Shape& shape);
/external/tensorflow/tensorflow/compiler/xla/service/
Dlocal_service.cc93 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(argument_shape)); in CompileExecutable()
Dtransfer_manager.cc191 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(on_host_shape)); in AllocateShapedBuffer()
Dhlo_evaluator.cc895 TF_CHECK_OK(ShapeUtil::ValidateShape(lhs_shape)); in HandleConvolution()
896 TF_CHECK_OK(ShapeUtil::ValidateShape(rhs_shape)); in HandleConvolution()
2109 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(instruction->shape())); in Evaluate()
2146 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(instruction->shape())); in Evaluate()
Dservice.cc232 return ShapeUtil::ValidateShape(shape_with_layout); in ValidateResultShapeWithLayout()
1239 TF_DCHECK_OK(ShapeUtil::ValidateShape(program_shape.result())); in ComputeConstant()
Dlayout_assignment.cc787 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(instruction->shape())); in CheckLayouts()
/external/tensorflow/tensorflow/core/graph/
Dgraph_constructor.cc190 Status ValidateShape(Node* node);
558 Status GraphConstructor::ValidateShape(Node* node) { in ValidateShape() function in tensorflow::__anond2858b560111::GraphConstructor
1016 TF_RETURN_IF_ERROR(ValidateShape(node)); in Convert()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc91 TF_CHECK_OK(ShapeUtil::ValidateShape(shape)); in IrArray()
Dllvm_util.cc247 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(shape)); in DecodeSelfDescribingShapeConstant()