Searched refs:ValidateShapeWithOptionalLayout (Results 1 – 13 of 13) sorted by relevance
244 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in InferUnaryOpShape()739 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(result)); in InferDotOpShape()970 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(lhs)); in InferBinaryOpShape()971 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(rhs)); in InferBinaryOpShape()1054 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(lhs)); in InferTernaryOpShape()1055 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(rhs)); in InferTernaryOpShape()1056 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(ehs)); in InferTernaryOpShape()1080 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(*shape)); in InferVariadicOpShape()1219 TF_RET_CHECK(ShapeUtil::ValidateShapeWithOptionalLayout(operand_shape) == in InferBatchNormTrainingShape()1221 TF_RET_CHECK(ShapeUtil::ValidateShapeWithOptionalLayout(offset_shape) == in InferBatchNormTrainingShape()[all …]
32 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(client_shape)); in ValidateResultShape()
117 ShapeUtil::ValidateShapeWithOptionalLayout(argument_shape)); in GetHloModuleConfig()
1815 ShapeUtil::ValidateShapeWithOptionalLayout(result_layout.shape())); in VerifyEntryComputationLayout()1836 ShapeUtil::ValidateShapeWithOptionalLayout(layout.parameter_shape(i))); in VerifyEntryComputationLayout()
224 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(client_shape)); in ValidateResultShape()
157 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in CreateFromProto()464 ShapeUtil::ValidateShapeWithOptionalLayout(outfeed_shape)); in CreateFromProto()3469 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape_)); in HloInstruction()
1526 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape())); in infeed_shape()
2033 ShapeUtil::ValidateShapeWithOptionalLayout(instruction->shape())); in AssignLayouts()
147 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(result)); in MakeTupleShapeImpl()422 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(result)); in MakeOpaqueShape()429 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(result)); in MakeTokenShape()435 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(shape)); in AppendShapeToTuple()566 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(shape.tuple_shapes(index))); in GetTupleElementShape()579 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(tuple)); in SliceTuple()945 /* static */ Status ShapeUtil::ValidateShapeWithOptionalLayout( in ValidateShapeWithOptionalLayout() function in xla::ShapeUtil
394 static Status ValidateShapeWithOptionalLayout(const Shape& shape);
470 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in CreateFromProto()
47 xla::ShapeUtil::ValidateShapeWithOptionalLayout(client_shape)); in ValidateResultShape()
2298 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in RngOp()2324 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in RngBitGenerator()