| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | symbolic_shapes.h | 26 bool IsKnown(const TensorShapeProto::Dim& dim);
27 bool IsKnownSymbolically(const TensorShapeProto::Dim& dim);
28 bool IsUnknown(const TensorShapeProto::Dim& dim);
32 bool ShapeIsSymbolicallyDefined(const TensorShapeProto& shape);
36 int Rank(const TensorShapeProto& shape);
41 int64 NumCoefficients(const TensorShapeProto& shape);
45 bool ShapesSymbolicallyEqual(const TensorShapeProto& left,
46 const TensorShapeProto& right);
52 bool ShapesBroadcastable(const TensorShapeProto& left,
53 const TensorShapeProto& right);
[all …]
|
| D | symbolic_shapes.cc | 26 BCast::Vec ShapeDims(const TensorShapeProto& shape) { in ShapeDims()
36 bool IsKnown(const TensorShapeProto::Dim& dim) { return dim.size() >= 0; } in IsKnown()
38 bool IsKnownSymbolically(const TensorShapeProto::Dim& dim) { in IsKnownSymbolically()
42 bool IsUnknown(const TensorShapeProto::Dim& dim) { return dim.size() == -1; } in IsUnknown()
44 bool ShapeIsSymbolicallyDefined(const TensorShapeProto& shape) { in ShapeIsSymbolicallyDefined()
48 [](const TensorShapeProto::Dim& dim) { return !IsUnknown(dim); }); in ShapeIsSymbolicallyDefined()
55 int Rank(const TensorShapeProto& shape) { in Rank()
62 int64 NumCoefficients(const TensorShapeProto& shape) { in NumCoefficients()
76 bool ShapesSymbolicallyEqual(const TensorShapeProto& left, in ShapesSymbolicallyEqual()
77 const TensorShapeProto& right) { in ShapesSymbolicallyEqual()
[all …]
|
| D | symbolic_shapes_test.cc | 26 TensorShapeProto MakeUnknown() { in MakeUnknown()
27 TensorShapeProto shape; in MakeUnknown()
32 TensorShapeProto MakeShape(std::vector<int> dims) { in MakeShape()
33 TensorShapeProto shape; in MakeShape()
35 TensorShapeProto::Dim dim; in MakeShape()
43 bool operator<(const TensorShapeProto& lhs, const TensorShapeProto& rhs) { in operator <()
78 TensorShapeProto output_shape; in TEST_F()
118 TensorShapeProto shape; in TEST_F()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | ragged_to_dense_util_test.cc | 30 TensorShapeProto shape_proto; in TEST()
32 TensorShapeProto value_shape_proto; in TEST()
36 TensorShapeProto actual_output_shape_proto; in TEST()
44 TensorShapeProto shape_proto; in TEST()
46 TensorShapeProto value_shape_proto; in TEST()
50 TensorShapeProto actual_output_shape_proto; in TEST()
60 TensorShapeProto shape_proto; in TEST()
62 TensorShapeProto value_shape_proto; in TEST()
67 TensorShapeProto actual_output_shape_proto; in TEST()
99 TensorShapeProto default_value_shape_proto; in TEST()
[all …]
|
| D | ragged_to_dense_util.h | 49 const TensorShapeProto& shape,
50 const TensorShapeProto& value_shape,
51 TensorShapeProto* output_shape);
55 Status ValidateDefaultValueShape(const TensorShapeProto& default_value_shape,
56 const TensorShapeProto& value_shape);
|
| D | ragged_to_dense_util.cc | 41 int ragged_rank, const TensorShapeProto& shape, in CombineRaggedTensorToTensorShapes()
42 const TensorShapeProto& value_shape, TensorShapeProto* output_shape) { in CombineRaggedTensorToTensorShapes()
74 const TensorShapeProto::Dim& value_dim = value_shape.dim(i); in CombineRaggedTensorToTensorShapes()
75 TensorShapeProto::Dim* output_shape_dim = output_shape->mutable_dim( in CombineRaggedTensorToTensorShapes()
96 const TensorShapeProto& default_value_shape, in ValidateDefaultValueShape()
97 const TensorShapeProto& value_shape) { in ValidateDefaultValueShape()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel.h | 125 const TensorShapeProto& tensor_shape,
134 const TensorShapeProto& MaxMemoryShape(const Node* node,
189 static Bytes MinTensorMemoryUsage(const TensorShapeProto& tensor_shape,
223 gtl::InlinedVector<TensorShapeProto, 2> output_port_shape;
233 TensorShapeProto unknown_shape_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape_test.cc | 122 TensorShapeProto proto; in TEST()
141 TensorShapeProto proto; in TEST()
248 explicit TensorShapeOld(const TensorShapeProto& proto);
255 static bool IsValid(const TensorShapeProto& proto);
259 static Status IsValidShape(const TensorShapeProto& proto);
313 void AsProto(TensorShapeProto* proto) const;
333 static string DebugString(const TensorShapeProto& proto);
377 bool TensorShapeOld::IsValid(const TensorShapeProto& proto) { in IsValid()
387 Status TensorShapeOld::IsValidShape(const TensorShapeProto& proto) { in IsValidShape()
405 TensorShapeOld::TensorShapeOld(const TensorShapeProto& proto) { in TensorShapeOld()
[all …]
|
| D | tensor_shape.proto | 13 message TensorShapeProto { message
19 // that work with TensorShapeProto may fail at runtime when deserializing
20 // a TensorShapeProto containing a dim value of -1.
|
| D | tensor_shape.h | 39 class TensorShapeProto; variable
75 static std::string DebugString(const TensorShapeProto& proto);
178 TensorShapeBase(const TensorShapeProto& proto);
190 static Status BuildTensorShapeBase(const TensorShapeProto& proto,
195 static bool IsValid(const TensorShapeProto& proto);
199 static Status IsValidShape(const TensorShapeProto& proto);
312 void AsProto(TensorShapeProto* proto) const;
|
| /external/tensorflow/tensorflow/go/ |
| D | signature_test.go | 36 TensorShape: &tspb.TensorShapeProto{
49 TensorShape: &tspb.TensorShapeProto{
64 TensorShape: &tspb.TensorShapeProto{
77 TensorShape: &tspb.TensorShapeProto{
148 TensorShape: &tspb.TensorShapeProto{
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.h | 31 TensorShapeProto* tensor_shape_proto);
32 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape,
288 const TensorShapeProto& original_image_shape,
289 const TensorShapeProto& original_filter_shape, const OpInfo& op_info,
294 const TensorShapeProto& original_image_shape, const OpInfo& op_info,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer.cc | 106 const TensorShapeProto& reduction_indices_shape = in Optimize()
169 const TensorShapeProto& shape1 = prop1[0].shape(); in Optimize()
170 const TensorShapeProto& shape2 = prop2[0].shape(); in Optimize()
180 *t->mutable_tensor_shape() = TensorShapeProto(); in Optimize()
|
| D | constant_folding.h | 124 const TensorShapeProto& shape,
232 TensorShapeProto* input_tensor_shape,
233 TensorShapeProto* output_tensor_shape, bool* is_single_element_op) const;
237 const NodeDef& node, const TensorShapeProto& input_shape, bool keep_dims,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_conversion_ops.cc | 72 TensorShapeProto partition_shape; in ValidateRowPartitionTypesAndShapes()
243 TensorShapeProto shape; in RaggedTensorToTensorShapeFn()
257 TensorShapeProto value_shape; in RaggedTensorToTensorShapeFn()
260 TensorShapeProto default_value_shape; in RaggedTensorToTensorShapeFn()
274 TensorShapeProto output_shape; in RaggedTensorToTensorShapeFn()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_output_layout_utils.cc | 24 std::vector<TensorShapeProto>* shapes) { in ComputeOutputTensorShapes()
29 TensorShapeProto shape; in ComputeOutputTensorShapes()
|
| /external/tensorflow/tensorflow/python/saved_model/model_utils/ |
| D | export_output_test.py | 79 shape = tensor_shape_pb2.TensorShapeProto(
80 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
112 shape = tensor_shape_pb2.TensorShapeProto(
113 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
148 shape = tensor_shape_pb2.TensorShapeProto(
149 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
189 shape = tensor_shape_pb2.TensorShapeProto(
190 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_type.h | 45 Status ConvertToMlirShape(const TensorShapeProto& input_shape,
49 StatusOr<mlir::Type> ConvertToMlirTensorType(const TensorShapeProto& shape,
|
| D | mangling_util.h | 44 string MangleShape(const TensorShapeProto& shape);
46 Status DemangleShape(absl::string_view str, TensorShapeProto* proto);
|
| D | convert_tensor.h | 43 TensorShapeProto* output_shape);
52 StatusOr<mlir::Attribute> ConvertTensorShapeProto(const TensorShapeProto& shape,
|
| D | mangling_util.cc | 63 string MangleShape(const TensorShapeProto& shape) { in MangleShape()
67 Status DemangleShape(absl::string_view str, TensorShapeProto* proto) { in DemangleShape()
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_svdf_test.cc | 106 tensorflow::TensorShapeProto* allocated_tensor_shape = in AddNewNode()
107 new tensorflow::TensorShapeProto; in AddNewNode()
125 tensorflow::TensorShapeProto* allocated_tensor_shape = in AddShapeNode()
126 new tensorflow::TensorShapeProto; in AddShapeNode()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | cpp_shape_inference.proto | 12 TensorShapeProto shape = 1;
25 TensorShapeProto shape = 1;
|
| /external/tensorflow/tensorflow/c/experimental/gradients/tape/ |
| D | tape_operation.cc | 95 TensorShapeProto proto; in SetAttrShape()
166 std::unique_ptr<TensorShapeProto[]> proto(new TensorShapeProto[num_values]); in SetAttrShapeList()
187 attr_name, gtl::ArraySlice<TensorShapeProto>(proto.get(), num_values)); in SetAttrShapeList()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | accumulate_n_optimizer.cc | 26 Tensor make_zeros(const DataType& dtype, const TensorShapeProto& shape) { in make_zeros()
128 TensorShapeProto shape; in RewriteIntoTempVariable()
149 TensorShapeProto variable_shape; in RewriteIntoTempVariable()
166 .Attr("value", make_zeros(dtype, TensorShapeProto())) in RewriteIntoTempVariable()
|