Home
last modified time | relevance | path

Searched refs:TF_DCHECK_OK (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/platform/
Dstatus.h228 #define TF_DCHECK_OK(val) TF_CHECK_OK(val) macro
230 #define TF_DCHECK_OK(val) \ macro
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc372 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(result)); in MakeTupleShape()
379 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(result)); in MakeOpaqueShape()
386 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(result)); in MakeTokenShape()
392 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(shape)); in AppendShapeToTuple()
420 TF_DCHECK_OK(ValidateShape(*shape)); in AppendMajorDimension()
429 TF_DCHECK_OK(ValidateShape(*to)); in CopyDynamicDimensions()
504 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(shape.tuple_shapes(index))); in GetTupleElementShape()
517 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(tuple)); in SliceTuple()
744 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOf()
761 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOfTupleIndexTable()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dgraph.cc494 TF_DCHECK_OK(IsValidNode(node)) << node->DebugString(); in RemoveNode()
517 TF_DCHECK_OK(IsValidNode(source)) << source->DebugString(); in AddEdge()
518 TF_DCHECK_OK(IsValidNode(dest)) << dest->DebugString(); in AddEdge()
548 TF_DCHECK_OK(IsValidNode(e->src_)) << e->src_->DebugString(); in RemoveEdge()
549 TF_DCHECK_OK(IsValidNode(e->dst_)) << e->dst_->DebugString(); in RemoveEdge()
820 TF_DCHECK_OK(IsValidNode(node)) << node->DebugString(); in ReleaseNode()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc264 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape)); in InferUnaryOpShape()
735 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(result)); in InferDotOpShape()
966 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(lhs)); in InferBinaryOpShape()
967 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(rhs)); in InferBinaryOpShape()
1050 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(lhs)); in InferTernaryOpShape()
1051 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(rhs)); in InferTernaryOpShape()
1052 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(ehs)); in InferTernaryOpShape()
1078 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(*shape)); in InferVariadicOpShape()
1677 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(lhs)); in InferConvolveShape()
1678 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(rhs)); in InferConvolveShape()
Dhlo_ordering.cc498 TF_DCHECK_OK(schedule_.Verify()); in Initialize()
Dcopy_insertion.cc537 TF_DCHECK_OK(Verify()); in CopyRemover()
808 TF_DCHECK_OK(Verify()); in TryElideCopy()
Dhlo_alias_analysis.cc559 TF_DCHECK_OK(alias_analysis->Verify()); in Run()
Dhlo_dataflow_analysis.cc1156 TF_DCHECK_OK(dataflow_analysis->Verify()); in Run()
Dservice.cc1064 TF_DCHECK_OK(ShapeUtil::ValidateShape(program_shape.result())); in ComputeConstantGraph()
Dhlo_instructions.h1168 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape())); in infeed_shape()
Dhlo_instruction.cc3052 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape_)); in HloInstruction()
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc1712 TF_DCHECK_OK(added); in ReachableFunctionLibraryDefinition()
1721 TF_DCHECK_OK(added_grad); in ReachableFunctionLibraryDefinition()