| /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 | 23 BCast::Vec ShapeDims(const TensorShapeProto& shape) { in ShapeDims()
33 bool IsKnown(const TensorShapeProto::Dim& dim) { return dim.size() >= 0; } in IsKnown()
35 bool IsKnownSymbolically(const TensorShapeProto::Dim& dim) { in IsKnownSymbolically()
39 bool IsUnknown(const TensorShapeProto::Dim& dim) { return dim.size() == -1; } in IsUnknown()
41 bool ShapeIsSymbolicallyDefined(const TensorShapeProto& shape) { in ShapeIsSymbolicallyDefined()
45 [](const TensorShapeProto::Dim& dim) { return !IsUnknown(dim); }); in ShapeIsSymbolicallyDefined()
52 int Rank(const TensorShapeProto& shape) { in Rank()
59 int64 NumCoefficients(const TensorShapeProto& shape) { in NumCoefficients()
73 bool ShapesSymbolicallyEqual(const TensorShapeProto& left, in ShapesSymbolicallyEqual()
74 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/framework/ |
| D | tensor_shape_test.cc | 121 TensorShapeProto proto; in TEST()
140 TensorShapeProto proto; in TEST()
225 explicit TensorShapeOld(const TensorShapeProto& proto);
232 static bool IsValid(const TensorShapeProto& proto);
236 static Status IsValidShape(const TensorShapeProto& proto);
290 void AsProto(TensorShapeProto* proto) const;
310 static string DebugString(const TensorShapeProto& proto);
354 bool TensorShapeOld::IsValid(const TensorShapeProto& proto) { in IsValid()
364 Status TensorShapeOld::IsValidShape(const TensorShapeProto& proto) { in IsValidShape()
382 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 | 37 class TensorShapeProto; variable
73 static string DebugString(const TensorShapeProto& proto);
177 TensorShapeBase(const TensorShapeProto& proto);
181 static bool IsValid(const TensorShapeProto& proto);
185 static Status IsValidShape(const TensorShapeProto& proto);
251 void AsProto(TensorShapeProto* proto) const;
|
| D | attr_value_util.h | 64 void SetAttrValue(const TensorShapeProto& value, AttrValue* out);
82 void SetAttrValue(gtl::ArraySlice<TensorShapeProto> value, AttrValue* out);
|
| D | tensor_shape.cc | 55 bool TensorShapeBase<Shape>::IsValid(const TensorShapeProto& proto) { in IsValid()
75 Status TensorShapeBase<Shape>::IsValidShape(const TensorShapeProto& proto) { in IsValidShape()
117 TensorShapeBase<Shape>::TensorShapeBase(const TensorShapeProto& proto) { in TensorShapeBase()
466 void TensorShapeBase<Shape>::AsProto(TensorShapeProto* proto) const { in AsProto()
527 string TensorShapeRep::DebugString(const TensorShapeProto& proto) { in DebugString()
595 TensorShapeProto proto; in MakeShapeHelper()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.h | 28 TensorShapeProto* tensor_shape_proto);
29 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape,
165 const TensorShapeProto& original_image_shape,
166 const TensorShapeProto& original_filter_shape, const OpInfo& op_info,
171 const TensorShapeProto& original_image_shape, const OpInfo& op_info,
|
| D | op_level_cost_estimator.cc | 152 int64 CwiseOutputElementCount(const TensorShapeProto& input_shape_1, in CwiseOutputElementCount()
153 const TensorShapeProto& input_shape_2) { in CwiseOutputElementCount()
156 TensorShapeProto output_shape = in CwiseOutputElementCount()
183 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape, in MaybeGetMinimumShape()
573 const TensorShapeProto& original_image_shape, in ConvolutionDimensionsFromInputs()
574 const TensorShapeProto& original_filter_shape, const OpInfo& op_info, in ConvolutionDimensionsFromInputs()
790 TensorShapeProto a_input_shape = MaybeGetMinimumShape( in CountBatchMatMulOperations()
793 TensorShapeProto b_input_shape = MaybeGetMinimumShape( in CountBatchMatMulOperations()
804 TensorShapeProto* bigger_rank_shape = &a_input_shape; in CountBatchMatMulOperations()
805 TensorShapeProto* smaller_rank_shape = &b_input_shape; in CountBatchMatMulOperations()
[all …]
|
| D | op_level_cost_estimator_test.cc | 926 const TensorShapeProto& tensor_shape_proto) { in ExpectTensorShape()
937 TensorShapeProto tensor_shape_proto; in TEST_F()
1255 TensorShapeProto x; in TEST_F()
1258 TensorShapeProto y = MaybeGetMinimumShape(x, 4, &unknown_shapes); in TEST_F()
1264 TensorShapeProto x; in TEST_F()
1267 TensorShapeProto y = MaybeGetMinimumShape(x, 1, &unknown_shapes); in TEST_F()
1273 TensorShapeProto x; in TEST_F()
1276 TensorShapeProto y = MaybeGetMinimumShape(x, 2, &unknown_shapes); in TEST_F()
1282 TensorShapeProto x; in TEST_F()
1287 TensorShapeProto y = MaybeGetMinimumShape(x, 2, &unknown_shapes); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel.h | 117 const TensorShapeProto& tensor_shape,
126 const TensorShapeProto& MaxMemoryShape(const Node* node,
181 static Bytes MinTensorMemoryUsage(const TensorShapeProto& tensor_shape,
215 gtl::InlinedVector<TensorShapeProto, 2> output_port_shape;
225 TensorShapeProto unknown_shape_;
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops_test.cc | 205 typedef std::vector<std::pair<TensorShapeProto, DataType>> ShapeDtypeV; in TEST()
212 {TensorShapeProto(), TensorShapeProto(), TensorShapeProto()}, {}, {}, in TEST()
220 TensorShapeProto p; in TEST()
225 TensorShapeProto i0 = shape_proto({1, -1}); in TEST()
226 TensorShapeProto i1 = shape_proto({-1, 2}); in TEST()
227 TensorShapeProto unknown_shape; in TEST()
229 TensorShapeProto scalar; in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer.cc | 68 const TensorShapeProto& reduction_indices_shape = in Optimize()
110 const TensorShapeProto& shape1 = prop1[0].shape(); in Optimize()
111 const TensorShapeProto& shape2 = prop2[0].shape(); in Optimize()
121 *t->mutable_tensor_shape() = TensorShapeProto(); in Optimize()
|
| D | constant_folding.h | 98 const TensorShapeProto& shape,
159 TensorShapeProto* input_tensor_shape,
160 TensorShapeProto* output_tensor_shape, bool* is_single_element_op) const;
165 const NodeDef& node, const TensorShapeProto& input_shape, bool keep_dims,
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | cpp_shape_inference.cc | 34 TensorShapeProto* out) { in ProtoFromShapeHandle()
75 std::vector<TensorShapeProto> input_shapes; in RunCppShapeInferenceImpl()
77 std::unique_ptr<std::vector<std::pair<TensorShapeProto, DataType>>>> in RunCppShapeInferenceImpl()
93 new std::vector<std::pair<TensorShapeProto, DataType>>); in RunCppShapeInferenceImpl()
116 std::vector<TensorShapeProto> input_tensor_as_shapes_protos( in RunCppShapeInferenceImpl()
|
| D | cpp_shape_inference.proto | 11 TensorShapeProto shape = 1;
20 TensorShapeProto shape = 1;
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| D | saved_model_export_utils_test.py | 89 shape = tensor_shape_pb2.TensorShapeProto(
90 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
120 shape = tensor_shape_pb2.TensorShapeProto(
121 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
160 shape = tensor_shape_pb2.TensorShapeProto(
161 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
206 shape = tensor_shape_pb2.TensorShapeProto(
207 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
249 shape = tensor_shape_pb2.TensorShapeProto(
250 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
[all …]
|
| /external/tensorflow/tensorflow/python/saved_model/model_utils/ |
| D | export_output_test.py | 78 shape = tensor_shape_pb2.TensorShapeProto(
79 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
111 shape = tensor_shape_pb2.TensorShapeProto(
112 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
147 shape = tensor_shape_pb2.TensorShapeProto(
148 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
188 shape = tensor_shape_pb2.TensorShapeProto(
189 dim=[tensor_shape_pb2.TensorShapeProto.Dim(size=1)])
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | accumulate_n_optimizer.cc | 22 Tensor make_zeros(const DataType& dtype, const TensorShapeProto& shape) { in make_zeros()
96 TensorShapeProto shape; in rewriteNode()
117 TensorShapeProto variable_shape; in rewriteNode()
134 .Attr("value", make_zeros(dtype, TensorShapeProto())) in rewriteNode()
|
| /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/contrib/predictor/ |
| D | saved_model_predictor_test.py | 95 tensor_shape=tensor_shape_pb2.TensorShapeProto()),
99 tensor_shape=tensor_shape_pb2.TensorShapeProto())}
104 tensor_shape=tensor_shape_pb2.TensorShapeProto())}
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_output_layout_utils.cc | 60 std::vector<TensorShapeProto>* shapes) { in ComputeOutputTensorShapes()
70 TensorShapeProto shape; in ComputeOutputTensorShapes()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.cc | 265 std::vector<int64> ShapeProtoToVec(const TensorShapeProto& shape_pb) { in ShapeProtoToVec()
278 TensorShapeProto VecToShapeProto(const std::vector<int64>& shape_vec) { in VecToShapeProto()
279 TensorShapeProto shape_pb; in VecToShapeProto()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-attr-value.pbtxt | 46 type_name: ".tensorflow.TensorShapeProto"
130 type_name: ".tensorflow.TensorShapeProto"
|