| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape.cc | 177 uint16* dst = as16()->dims_; in InitDims()
244 delete as64()->dims_; in DestructorOutOfLine()
250 delete as64()->dims_; in SlowCopyFrom()
263 *(as64()->dims_) = *(b.as64()->dims_); in SlowCopyFrom()
266 as64()->dims_ = new gtl::InlinedVector<int64, 4>(*(b.as64()->dims_)); in SlowCopyFrom()
277 uint16 dim = as16()->dims_[d]; in dim_size()
281 uint32 dim = as32()->dims_[d]; in dim_size()
285 return (*as64()->dims_)[d]; in dim_size()
296 delete as64()->dims_; in ClearAllButDataType()
341 as16()->dims_[nd] = in UnsafeAddDim()
[all …]
|
| D | shape_inference.h | 90 const std::vector<DimensionHandle> dims_; variable
352 return s->dims_[s->dims_.size() + idx]; in DimKnownRank()
354 return s->dims_[idx]; in DimKnownRank()
804 : rank_(dims.size()), dims_(dims) {} in Shape()
|
| D | tensor_shape.h | 95 uint16 dims_[6]; member
98 uint32 dims_[3]; member
101 gtl::InlinedVector<int64, 4>* dims_; member
|
| D | shape_inference.cc | 336 for (auto d : s->dims_) vals.push_back(DebugString(d)); in DebugString()
718 dim_index = s->dims_.size() + dim_index; in ReplaceDim()
720 if (!FastBoundsCheck(dim_index, s->dims_.size())) { in ReplaceDim()
723 " for shape with ", s->dims_.size(), in ReplaceDim()
726 std::vector<DimensionHandle> dims(s->dims_); in ReplaceDim()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 97 SparseTensor() : dims_(0) {} in SparseTensor()
120 dims_(UnsafeGetDimsFromIx(ix)) { in SparseTensor()
127 DCHECK_EQ(order.size(), dims_) << "Order length must be SparseTensor rank."; in SparseTensor()
128 DCHECK_EQ(shape.size(), dims_) << "Shape rank must be SparseTensor rank."; in SparseTensor()
143 dims_ = other.dims_;
152 dims_ = std::move(other.dims_);
198 DCHECK_LE(group_ix.size(), dims_); in group()
201 DCHECK_LT(group_ix[di], dims_) << "Group dimension out of range"; in group()
205 return GroupIterable(ix_, vals_, dims_, group_ix); in group()
292 for (int di = 0; di < dims_; ++di) { in IndexValid()
[all …]
|
| D | dim_comparator.h | 51 : ix_(ix), order_(order), dims_(shape.size()) { in DimComparator()
61 for (int di = 0; di < dims_; ++di) { in operator()
90 const int dims_; variable
|
| D | group_iterator.h | 84 dims_(dims), in GroupIterable()
134 const int dims_; variable
|
| D | group_iterator.cc | 66 next_loc_ - loc_, iter_->dims_); in indices()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 327 Shape(std::initializer_list<int> dim_list) : dims_(dim_list) {}
330 dims_ = std::vector<int>(dim_list);
333 const std::vector<int>& dims() const { return dims_; }
334 std::vector<int>* mutable_dims() { return &dims_; }
335 const int dimensions_count() const { return dims_.size(); }
344 CHECK_GT(dims_.size(), i);
345 return dims_[i];
349 return (this->dims_ == comp.dims());
355 std::vector<int> dims_;
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 177 return size_ > kMaxSmallSize ? dims_pointer_[i] : dims_[i]; in Dims()
185 dims_[i] = val; in SetDim()
190 return size_ > kMaxSmallSize ? dims_pointer_ : dims_; in DimsData()
193 return size_ > kMaxSmallSize ? dims_pointer_ : dims_; in DimsData()
196 inline const int32* DimsDataUpTo4D() const { return dims_; } in DimsDataUpTo4D()
274 int32 dims_[kMaxSmallSize]; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 184 return size_ > kMaxSmallSize ? dims_pointer_[i] : dims_[i]; in Dims()
192 dims_[i] = val; in SetDim()
197 return size_ > kMaxSmallSize ? dims_pointer_ : dims_; in DimsData()
200 return size_ > kMaxSmallSize ? dims_pointer_ : dims_; in DimsData()
203 inline const int32* DimsDataUpTo4D() const { return dims_; } in DimsDataUpTo4D()
290 int32 dims_[kMaxSmallSize]; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 75 dims_(shape.dimensions().begin(), shape.dimensions().end()) { in Index()
89 dims_(shape.dimensions().begin(), shape.dimensions().end()), in Index()
123 auto b = ShapeUtil::MakeShape(a.element_type(), dims_); in LinearValidOnShape()
|
| D | ir_array.h | 198 std::vector<int64> dims_; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 661 dims_(dims), in Config()
676 Dimensions dims() const { return dims_; } in dims()
686 Dimensions dims_; member in xla::cpu::__anon016ac5180111::TiledSmallGemmEmitter::Config
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1724 TF_RETURN_IF_ERROR(dims_.Merge(InferenceContext::DimKnownRank(s1, i), in Merge()
1734 return dims_.Merge(d1, d2); in Merge()
1747 int64 d = dims_.GetMergedValue(dim); in AsTensorProperties()
1755 DisjointSet<shape_inference::DimensionHandle> dims_; member in tensorflow::grappler::SymbolicShapeManager
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 210 FakeITensor(const nvinfer1::Dims& dims) : dims_(dims), dynamic_range_(0.0f) {} in FakeITensor()
213 : dims_(GetTestDims(dims)), dynamic_range_(0.0f) {} in FakeITensor()
219 void setDimensions(nvinfer1::Dims dimensions) override { dims_ = dimensions; } in setDimensions()
221 nvinfer1::Dims getDimensions() const override { return dims_; } in getDimensions()
262 nvinfer1::Dims dims_; member in tensorflow::tensorrt::convert::FakeITensor
|