Searched refs:ValidateShapeWithOptionalLayout (Results 1 – 13 of 13) sorted by relevance
240 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 …]
137 ShapeUtil::ValidateShapeWithOptionalLayout(argument_shape)); in CompileExecutable()
948 ShapeUtil::ValidateShapeWithOptionalLayout(result_layout.shape())); in VerifyEntryComputationLayout()971 ShapeUtil::ValidateShapeWithOptionalLayout(layout.parameter_shape(i))); in VerifyEntryComputationLayout()
134 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()
218 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(client_shape)); in ValidateResultShape()
976 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape())); in infeed_shape()
1657 ShapeUtil::ValidateShapeWithOptionalLayout(instruction->shape())); in AssignLayouts()
28 ShapeUtil::ValidateShapeWithOptionalLayout(instruction->shape())); in Run()
244 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
433 static Status ValidateShapeWithOptionalLayout(const Shape& shape);
303 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in CreateFromProto()
1800 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in RngOp()