| /external/tensorflow/tensorflow/core/util/sparse/ |
| dim_comparator.h | 51 : ix_(ix), order_(order), dims_(shape.size()) {
61 for (int di = 0; di < dims_; ++di) {
90 const int dims_; member in class:tensorflow::sparse::DimComparator
|
| sparse_tensor.h | 97 SparseTensor() : dims_(0) {}
120 dims_(UnsafeGetDimsFromIx(ix)) {
127 DCHECK_EQ(order.size(), dims_) << "Order length must be SparseTensor rank.";
128 DCHECK_EQ(shape.size(), dims_) << "Shape rank must be SparseTensor rank.";
143 dims_ = other.dims_;
152 dims_ = std::move(other.dims_);
198 DCHECK_LE(group_ix.size(), dims_);
201 DCHECK_LT(group_ix[di], dims_) << "Group dimension out of range"
373 int dims_; member in class:tensorflow::sparse::SparseTensor
[all...] |
| group_iterator.h | 84 dims_(dims),
134 const int dims_; member in class:tensorflow::sparse::GroupIterable
|
| group_iterator.cc | 66 next_loc_ - loc_, iter_->dims_);
|
| /external/tensorflow/tensorflow/core/framework/ |
| tensor_shape.cc | 177 uint16* dst = as16()->dims_;
244 delete as64()->dims_;
250 delete as64()->dims_;
263 *(as64()->dims_) = *(b.as64()->dims_);
266 as64()->dims_ = new gtl::InlinedVector<int64, 4>(*(b.as64()->dims_));
277 uint16 dim = as16()->dims_[d];
281 uint32 dim = as32()->dims_[d];
285 return (*as64()->dims_)[d]
[all...] |
| shape_inference.h | 90 const std::vector<DimensionHandle> dims_; member in class:tensorflow::shape_inference::Shape
352 return s->dims_[s->dims_.size() + idx];
354 return s->dims_[idx];
804 : rank_(dims.size()), dims_(dims) {}
[all...] |
| tensor_shape.h | 95 uint16 dims_[6]; member in struct:tensorflow::TensorShapeRep::Rep16
98 uint32 dims_[3]; member in struct:tensorflow::TensorShapeRep::Rep32
101 gtl::InlinedVector<int64, 4>* dims_; member in struct:tensorflow::TensorShapeRep::Rep64
|
| shape_inference.cc | 336 for (auto d : s->dims_) vals.push_back(DebugString(d));
718 dim_index = s->dims_.size() + dim_index;
720 if (!FastBoundsCheck(dim_index, s->dims_.size())) {
723 " for shape with ", s->dims_.size(),
726 std::vector<DimensionHandle> dims(s->dims_);
[all...] |
| /external/tensorflow/tensorflow/lite/toco/ |
| model.h | 321 // we hide the raw dims_ member, but expose it raw by accessors
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_; member in class:toco::Shape
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| types.h | 177 return size_ > kMaxSmallSize ? dims_pointer_[i] : dims_[i];
185 dims_[i] = val;
190 return size_ > kMaxSmallSize ? dims_pointer_ : dims_;
193 return size_ > kMaxSmallSize ? dims_pointer_ : dims_;
196 inline const int32* DimsDataUpTo4D() const { return dims_; }
274 int32 dims_[kMaxSmallSize]; member in union:tflite::RuntimeShape::__anon44757
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| types.h | 184 return size_ > kMaxSmallSize ? dims_pointer_[i] : dims_[i];
192 dims_[i] = val;
197 return size_ > kMaxSmallSize ? dims_pointer_ : dims_;
200 return size_ > kMaxSmallSize ? dims_pointer_ : dims_;
203 inline const int32* DimsDataUpTo4D() const { return dims_; }
290 int32 dims_[kMaxSmallSize]; member in union:tflite::RuntimeShape::__anon46053
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| ir_array.cc | 75 dims_(shape.dimensions().begin(), shape.dimensions().end()) {
89 dims_(shape.dimensions().begin(), shape.dimensions().end()),
123 auto b = ShapeUtil::MakeShape(a.element_type(), dims_);
|
| ir_array.h | 189 // `dims_`, and if the `Index` is used in another array, and its layout and
195 // be null and `layout_` and `dims_` would be ignored.
198 std::vector<int64> dims_; member in class:xla::llvm_ir::IrArray::Index
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| tiled_dot_emitter.cc | 661 dims_(dims),
676 Dimensions dims() const { return dims_; }
686 Dimensions dims_; member in class:xla::cpu::__anon44367::TiledSmallGemmEmitter::Config
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_properties.cc | 1755 DisjointSet<shape_inference::DimensionHandle> dims_; member in class:tensorflow::grappler::SymbolicShapeManager
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| convert_nodes_test.cc | 210 FakeITensor(const nvinfer1::Dims& dims) : dims_(dims), dynamic_range_(0.0f) {}
213 : dims_(GetTestDims(dims)), dynamic_range_(0.0f) {}
219 void setDimensions(nvinfer1::Dims dimensions) override { dims_ = dimensions; }
221 nvinfer1::Dims getDimensions() const override { return dims_; }
262 nvinfer1::Dims dims_; member in class:tensorflow::tensorrt::convert::FakeITensor
[all...] |
