Home
last modified time | relevance | path

Searched refs:TensorShapeBase (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.cc36 static void AppendTo(const TensorShapeBase<Shape>& s, in AppendTo()
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() function in tensorflow::TensorShapeBase
136 TensorShapeBase<Shape>::TensorShapeBase(gtl::ArraySlice<int64> dim_sizes) { in TensorShapeBase() function in tensorflow::TensorShapeBase
159 void TensorShapeBase<Shape>::InitDims(gtl::ArraySlice<int64> dim_sizes) { in InitDims()
230 TensorShapeBase<Shape>::TensorShapeBase() { in TensorShapeBase() function in tensorflow::TensorShapeBase
272 int64 TensorShapeBase<Shape>::dim_size(int d) const { in dim_size()
305 void TensorShapeBase<Shape>::RecomputeNumElements() { in RecomputeNumElements()
323 void TensorShapeBase<Shape>::AddDim(int64 size) { in AddDim()
[all …]
Dtensor_shape.h166 class TensorShapeBase : public TensorShapeRep {
170 explicit TensorShapeBase(gtl::ArraySlice<int64> dim_sizes);
171 TensorShapeBase(std::initializer_list<int64> dim_sizes) in TensorShapeBase() function
172 : TensorShapeBase(gtl::ArraySlice<int64>(dim_sizes)) {} in TensorShapeBase()
175 TensorShapeBase();
177 TensorShapeBase(const TensorShapeProto& proto);
192 void AppendShape(const TensorShapeBase& shape);
277 std::ostream& operator<<(std::ostream& os, const TensorShapeBase<Shape>& tsb) {
290 class TensorShape : public TensorShapeBase<TensorShape> {
292 using TensorShapeBase<TensorShape>::TensorShapeBase;
[all …]