Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc240 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in InferUnaryOpShape()
705 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(result)); in InferDotOpShape()
922 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(lhs)); in InferBinaryOpShape()
923 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(rhs)); in InferBinaryOpShape()
1006 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(lhs)); in InferTernaryOpShape()
1007 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(rhs)); in InferTernaryOpShape()
1008 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(ehs)); in InferTernaryOpShape()
1034 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(*shape)); in InferVariadicOpShape()
1176 TF_RET_CHECK(ShapeUtil::ValidateShapeWithOptionalLayout(operand_shape) == in InferBatchNormTrainingShape()
1178 TF_RET_CHECK(ShapeUtil::ValidateShapeWithOptionalLayout(offset_shape) == in InferBatchNormTrainingShape()
[all …]
Dlocal_service.cc137 ShapeUtil::ValidateShapeWithOptionalLayout(argument_shape)); in CompileExecutable()
Dhlo_verifier.cc948 ShapeUtil::ValidateShapeWithOptionalLayout(result_layout.shape())); in VerifyEntryComputationLayout()
971 ShapeUtil::ValidateShapeWithOptionalLayout(layout.parameter_shape(i))); in VerifyEntryComputationLayout()
Dhlo_instruction.cc134 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in CreateFromProto()
370 ShapeUtil::ValidateShapeWithOptionalLayout(outfeed_shape)); in CreateFromProto()
2473 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape_)); in HloInstruction()
Dservice.cc218 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(client_shape)); in ValidateResultShape()
Dhlo_instructions.h976 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape())); in infeed_shape()
Dlayout_assignment.cc1657 ShapeUtil::ValidateShapeWithOptionalLayout(instruction->shape())); in AssignLayouts()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_hlo_support_checker.cc28 ShapeUtil::ValidateShapeWithOptionalLayout(instruction->shape())); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_hlo_support_checker.cc28 ShapeUtil::ValidateShapeWithOptionalLayout(instruction->shape())); in Run()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc244 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(result)); in MakeTupleShape()
251 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(result)); in MakeOpaqueShape()
258 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(result)); in MakeTokenShape()
264 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(shape)); in AppendShapeToTuple()
347 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(shape.tuple_shapes(index))); in GetTupleElementShape()
360 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(tuple)); in SliceTuple()
748 /* static */ Status ShapeUtil::ValidateShapeWithOptionalLayout( in ValidateShapeWithOptionalLayout() function in xla::ShapeUtil
Dshape_util.h433 static Status ValidateShapeWithOptionalLayout(const Shape& shape);
Dliteral.cc303 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in CreateFromProto()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1800 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in RngOp()