Home
last modified time | relevance | path

Searched refs:ValidateShapeWithOptionalLayout (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc240 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in InferUnaryOpShape()
715 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(result)); in InferDotOpShape()
946 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(lhs)); in InferBinaryOpShape()
947 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(rhs)); in InferBinaryOpShape()
1030 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(lhs)); in InferTernaryOpShape()
1031 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(rhs)); in InferTernaryOpShape()
1032 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(ehs)); in InferTernaryOpShape()
1058 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(*shape)); in InferVariadicOpShape()
1200 TF_RET_CHECK(ShapeUtil::ValidateShapeWithOptionalLayout(operand_shape) == in InferBatchNormTrainingShape()
1202 TF_RET_CHECK(ShapeUtil::ValidateShapeWithOptionalLayout(offset_shape) == in InferBatchNormTrainingShape()
[all …]
Dlocal_service.cc141 ShapeUtil::ValidateShapeWithOptionalLayout(argument_shape)); in CompileExecutables()
Dhlo_verifier.cc1141 ShapeUtil::ValidateShapeWithOptionalLayout(result_layout.shape())); in VerifyEntryComputationLayout()
1162 ShapeUtil::ValidateShapeWithOptionalLayout(layout.parameter_shape(i))); in VerifyEntryComputationLayout()
Dservice.cc219 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(client_shape)); in ValidateResultShape()
Dhlo_instruction.cc135 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in CreateFromProto()
380 ShapeUtil::ValidateShapeWithOptionalLayout(outfeed_shape)); in CreateFromProto()
2723 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape_)); in HloInstruction()
Dhlo_instructions.h1053 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape())); in infeed_shape()
Dlayout_assignment.cc1775 ShapeUtil::ValidateShapeWithOptionalLayout(instruction->shape())); in AssignLayouts()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc271 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(result)); in MakeTupleShape()
278 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(result)); in MakeOpaqueShape()
285 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(result)); in MakeTokenShape()
291 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(shape)); in AppendShapeToTuple()
394 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(shape.tuple_shapes(index))); in GetTupleElementShape()
407 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(tuple)); in SliceTuple()
748 /* static */ Status ShapeUtil::ValidateShapeWithOptionalLayout( in ValidateShapeWithOptionalLayout() function in xla::ShapeUtil
Dshape_util.h446 static Status ValidateShapeWithOptionalLayout(const Shape& shape);
Dliteral.cc296 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in CreateFromProto()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1738 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in RngOp()