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.h192 #define TF_DCHECK_OK(val) TF_CHECK_OK(val) macro
194 #define TF_DCHECK_OK(val) \ macro
/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()
319 TF_DCHECK_OK(ValidateShape(*shape)); in AppendMajorDimension()
394 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(shape.tuple_shapes(index))); in GetTupleElementShape()
407 TF_DCHECK_OK(ValidateShapeWithOptionalLayout(tuple)); in SliceTuple()
625 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOf()
643 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOfTupleIndexTable()
650 TF_DCHECK_OK(ValidateShape(shape)); in ByteSizeOfElements()
/external/tensorflow/tensorflow/core/graph/
Dgraph.cc469 TF_DCHECK_OK(IsValidNode(node)) << node->DebugString(); in RemoveNode()
492 TF_DCHECK_OK(IsValidNode(source)) << source->DebugString(); in AddEdge()
493 TF_DCHECK_OK(IsValidNode(dest)) << dest->DebugString(); in AddEdge()
523 TF_DCHECK_OK(IsValidNode(e->src_)) << e->src_->DebugString(); in RemoveEdge()
524 TF_DCHECK_OK(IsValidNode(e->dst_)) << e->dst_->DebugString(); in RemoveEdge()
795 TF_DCHECK_OK(IsValidNode(node)) << node->DebugString(); in ReleaseNode()
/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()
1661 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(lhs)); in InferConvolveShape()
1662 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(rhs)); in InferConvolveShape()
Dhlo_ordering.cc387 TF_DCHECK_OK(schedule_.Verify()); in Initialize()
Dcopy_insertion.cc518 TF_DCHECK_OK(Verify()); in CopyRemover()
768 TF_DCHECK_OK(Verify()); in TryElideCopy()
Dhlo_alias_analysis.cc525 TF_DCHECK_OK(alias_analysis->Verify()); in Run()
Dhlo_dataflow_analysis.cc966 TF_DCHECK_OK(dataflow_analysis->Verify()); in Run()
Dservice.cc1108 TF_DCHECK_OK(ShapeUtil::ValidateShape(program_shape.result())); in ComputeConstantGraph()
Dhlo_instructions.h1053 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape())); in infeed_shape()
Dhlo_instruction.cc2723 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(shape_)); in HloInstruction()
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc1670 TF_DCHECK_OK(added); in ReachableFunctionLibraryDefinition()
1679 TF_DCHECK_OK(added_grad); in ReachableFunctionLibraryDefinition()