| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | symbolic_shapes.cc | 42 return !shape.unknown_rank() && in ShapeIsSymbolicallyDefined()
53 if (shape.unknown_rank()) { in Rank()
60 if (shape.unknown_rank()) { in NumCoefficients()
75 if (left.unknown_rank() || right.unknown_rank() || in ShapesSymbolicallyEqual()
131 if (left.unknown_rank() || right.unknown_rank()) { in CompareSymbolicallyShapedTensorSizes()
218 if (numerator.unknown_rank() || denominator.unknown_rank()) { in ComputeSizeRatio()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape.cc | 59 if (kIsPartial && proto.unknown_rank()) return proto.dim_size() == 0; in IsValid()
79 if (kIsPartial && proto.unknown_rank()) { in IsValidShape()
123 if (kIsPartial && proto.unknown_rank()) { in TensorShapeBase()
273 if (unknown_rank()) return -1; in dim_size()
306 if (unknown_rank()) { in RecomputeNumElements()
325 if (unknown_rank()) return; in AddDim()
439 if (unknown_rank()) return; in RemoveDimRange()
468 if (unknown_rank()) { in AsProto()
506 CHECK(!unknown_rank()); in end()
512 if (shape.unknown_rank()) return "<unknown>"; in DebugString()
[all …]
|
| D | tensor_shape.proto | 39 // If "dim.size()" > 0, "unknown_rank" must be false.
45 bool unknown_rank = 3; field
|
| D | variant_test.cc | 114 EXPECT_EQ(x.get<TensorProto>()->tensor_shape().unknown_rank(), true); in TEST()
|
| D | partial_tensor_shape_test.cc | 31 EXPECT_TRUE(s.unknown_rank()); in TEST()
|
| D | tensor_shape.h | 225 bool unknown_rank() const { in unknown_rank() function
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 84 if (p.shape().unknown_rank()) { in PropToString()
141 EXPECT_FALSE(prop.shape().unknown_rank()); in TEST_F()
150 EXPECT_FALSE(in_prop.shape().unknown_rank()); in TEST_F()
217 EXPECT_TRUE(prop.shape().unknown_rank()); in TEST_F()
223 EXPECT_FALSE(prop.shape().unknown_rank()); in TEST_F()
268 EXPECT_FALSE(prop.shape().unknown_rank()); in TEST_F()
282 EXPECT_FALSE(prop.shape().unknown_rank()); in TEST_F()
313 EXPECT_FALSE(prop.shape().unknown_rank()); in TEST_F()
338 EXPECT_FALSE(prop.shape().unknown_rank()); in TEST_F()
1299 EXPECT_FALSE(out_prop0.shape().unknown_rank()); in TEST_F()
[all …]
|
| D | graph_memory.cc | 106 if (shape.unknown_rank()) { in InferMemUsageForNeighbors()
|
| D | utils.cc | 205 if (shape.unknown_rank()) { in CalculateTensorSize()
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_svdf.cc | 204 allocated_tensor_shape->set_unknown_rank(shape.unknown_rank()); in MaybeMergeConstNodes()
208 CHECK_EQ(allocated_tensor_shape->unknown_rank(), shape.unknown_rank()); in MaybeMergeConstNodes()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op.cc | 192 if (ts1.dims() != ts2.dims() || ts1.unknown_rank() || ts2.unknown_rank()) in MostSpecificCompatibleShape()
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/ |
| D | saved_model.pbtxt | 287 unknown_rank: true
874 unknown_rank: true
889 unknown_rank: true
2105 unknown_rank: true
2243 unknown_rank: true
2381 unknown_rank: true
2553 unknown_rank: true
2620 unknown_rank: true
2651 unknown_rank: true
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | custom_gradient.py | 70 ranks = [len(s.dim) if not s.unknown_rank else -1 for s in shapes]
72 if not s.unknown_rank else None for s in shapes]
|
| D | resource_variable_ops.py | 86 ranks = [len(s.dim) if not s.unknown_rank else -1 for s in shapes]
88 if not s.unknown_rank else None for s in shapes]
1503 if shape_proto.unknown_rank or any(x.size == -1 for x in shape_proto.dim):
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | directed_interleave_dataset_op.cc | 261 if (ts1.dims() != ts2.dims() || ts1.unknown_rank() || ts2.unknown_rank()) in MostSpecificCompatibleShape()
|
| D | unbatch_dataset_op.cc | 44 if (!shape.unknown_rank()) { in Dataset()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_shape_test.py | 405 self.assertTrue(tensor_shape.unknown_shape().as_proto().unknown_rank)
407 tensor_shape.unknown_shape(rank=3).as_proto().unknown_rank)
409 tensor_shape.TensorShape([1, 2, 3]).as_proto().unknown_rank)
411 tensor_shape.TensorShape([1, None, 3]).as_proto().unknown_rank)
|
| D | function_def_to_graph_test.py | 185 self.assertEqual(g.node[0].attr["shape"].shape.unknown_rank, False)
189 self.assertEqual(g.node[0].attr["shape"].shape.unknown_rank, False)
|
| D | tensor_shape.py | 721 if dims.unknown_rank:
1134 return tensor_shape_pb2.TensorShapeProto(unknown_rank=True)
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | model_analyzer.cc | 56 if (prop.shape().unknown_rank()) { in PrintNodeInfo()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 375 (op == "Rank" && shape.unknown_rank())) { in MaterializeShapes()
517 if (shp.unknown_rank()) { in ExtractShape()
687 if (input_prop.shape().unknown_rank()) { in MaterializeReductionIndices()
711 output_prop.shape().unknown_rank() ? -1 : output_prop.shape().dim_size(); in MaterializeReductionIndices()
1940 if (shape.unknown_rank()) { in RemoveShuffleOrTranspose()
1989 if (!shape.unknown_rank() && in RemoveRandomShuffle()
2005 if (shape.unknown_rank()) { in RemoveReverse()
2032 bool replaceable = !shape.unknown_rank(); in RemoveReverse()
2071 bool replaceable = !input.shape().unknown_rank(); in SimplifySlice()
2164 bool replaceable = !input.shape().unknown_rank(); in SimplifyStridedSlice()
[all …]
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model.py | 36 if val.tensor_shape.unknown_rank:
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 311 if (!element_shape.unknown_rank()) { in TensorListConcat()
337 OP_REQUIRES(c, element_shape.unknown_rank() || element_shape.dims() >= 1, in Compute()
507 OP_REQUIRES(c, element_shape.unknown_rank() || element_shape.dims() >= 1, in Compute()
521 if (!element_shape.unknown_rank()) { in Compute()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization.cc | 141 if (shape.unknown_rank()) return false; in IsOutputShapesFullyDefined()
232 if (!shape.unknown_rank()) { in AddNewBatchNode()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.cc | 267 if (shape_pb.dim_size() == 0 && !shape_pb.unknown_rank()) { in ShapeProtoToVec()
|