| /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_t 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_t 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.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()
|
| 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);
|
| /external/tensorflow/tensorflow/core/tpu/ops/ |
| D | tpu_embedding_shape_util.cc | 39 TensorShapeProto* shape) { in ComputeOneTableShape()
51 *shape = TensorShapeProto(); in ComputeOneTableShape()
65 std::vector<TensorShapeProto>* shapes) { in ComputeTableShapes()
77 std::vector<TensorShapeProto>* shapes) { in ComputeTableShapes()
88 TensorShapeProto TpuEmbeddingShapeUtil::MakeEmpty2DShape() { in MakeEmpty2DShape()
89 TensorShapeProto shape; in MakeEmpty2DShape()
|
| D | tpu_embedding_shape_util.h | 40 TensorShapeProto* shape);
50 std::vector<TensorShapeProto>* shapes);
54 int num_shards, std::vector<TensorShapeProto>* shapes);
56 static TensorShapeProto MakeEmpty2DShape();
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape.pb_text.h | 24 const ::tensorflow::TensorShapeProto& msg);
26 const ::tensorflow::TensorShapeProto& msg);
29 ::tensorflow::TensorShapeProto* msg)
|
| 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.h | 37 class TensorShapeProto; variable
73 static std::string DebugString(const TensorShapeProto& proto);
177 TensorShapeBase(const TensorShapeProto& proto);
189 static Status BuildTensorShapeBase(const TensorShapeProto& proto,
194 static bool IsValid(const TensorShapeProto& proto);
198 static Status IsValidShape(const TensorShapeProto& proto);
311 void AsProto(TensorShapeProto* proto) const;
312 TensorShapeProto AsProto() const;
374 static Status BuildTensorShape(const TensorShapeProto& proto, in BuildTensorShape()
379 static StatusOr<TensorShape> BuildTensorShape(const TensorShapeProto& proto) { in BuildTensorShape()
[all …]
|
| D | tensor_shape.pb_text.cc | 100 const ::tensorflow::TensorShapeProto& msg) { in ProtoDebugString()
109 const ::tensorflow::TensorShapeProto& msg) { in ProtoShortDebugString()
121 const ::tensorflow::TensorShapeProto& msg) { in AppendProtoDebugString()
134 ::tensorflow::TensorShapeProto* msg) { in ProtoParseFromString()
146 ::tensorflow::TensorShapeProto* msg) { in ProtoParseFromScanner()
|
| D | tensor_shape.pb_text-impl.h | 23 const ::tensorflow::TensorShapeProto& msg);
26 ::tensorflow::TensorShapeProto* msg);
|
| /external/federated-compute/fcp/aggregation/core/ |
| D | tensor_shape.cc | 41 const TensorShapeProto& shape_proto) { in FromProto()
54 TensorShapeProto TensorShape::ToProto() const { in ToProto()
55 TensorShapeProto shape_proto; in ToProto()
|
| /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/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/ir/importexport/ |
| D | convert_tensor.h | 43 tensorflow::TensorShapeProto* output_shape);
50 const tensorflow::TensorShapeProto& shape, MLIRContext* context);
59 tensorflow::TensorShapeProto* proto) { in SetTensorShapeProto()
|
| D | mangling.cc | 30 using tensorflow::TensorShapeProto;
70 std::string MangleShape(const TensorShapeProto& shape) { in MangleShape()
74 Status DemangleShape(absl::string_view str, TensorShapeProto* proto) { in DemangleShape()
|
| /external/pytorch/torch/utils/tensorboard/ |
| D | _proto_graph.py | 5 from tensorboard.compat.proto.tensor_shape_pb2 import TensorShapeProto
29 return TensorShapeProto(dim=[TensorShapeProto.Dim(size=d) for d in outputsize])
|
| D | summary.py | 20 from tensorboard.compat.proto.tensor_shape_pb2 import TensorShapeProto
420 "tensor_shape": TensorShapeProto(
421 dim=[TensorShapeProto.Dim(size=x) for x in tensor.shape]
753 tensor_shape=TensorShapeProto(),
770 tensor_shape=TensorShapeProto(dim=[TensorShapeProto.Dim(size=1)]),
793 tensor_shape=TensorShapeProto(
795 TensorShapeProto.Dim(size=data.shape[0]),
796 TensorShapeProto.Dim(size=data.shape[1]),
819 tensor_shape=TensorShapeProto(
821 TensorShapeProto.Dim(size=data.shape[0]),
[all …]
|
| D | _onnx_graph.py | 6 from tensorboard.compat.proto.tensor_shape_pb2 import TensorShapeProto
25 shapeproto = TensorShapeProto(
27 TensorShapeProto.Dim(size=d.dim_value)
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_output_layout_utils.cc | 25 std::vector<TensorShapeProto>* shapes) { in ComputeOutputTensorShapes()
33 TensorShapeProto shape; in ComputeOutputTensorShapes()
48 TensorShapeProto shape; in ComputeOutputTensorShapes()
|
| /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()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_conversion_ops.cc | 72 TensorShapeProto partition_shape; in ValidateRowPartitionTypesAndShapes()
240 TensorShapeProto shape; in RaggedTensorToTensorShapeFn()
254 TensorShapeProto value_shape; in RaggedTensorToTensorShapeFn()
257 TensorShapeProto default_value_shape; in RaggedTensorToTensorShapeFn()
271 TensorShapeProto output_shape; in RaggedTensorToTensorShapeFn()
|