| /art/tools/veridex/ |
| veridex.h | 50 : kind_(k), dimensions_(dims), class_def_(cl) {}
53 return kind_ == Primitive::Type::kPrimNot && dimensions_ == 0 && class_def_ == nullptr;
57 return kind_ != Primitive::Type::kPrimNot && dimensions_ == 0;
61 return dimensions_ != 0;
65 uint8_t GetDimensions() const { return dimensions_; }
95 uint8_t dimensions_; member in class:art::VeriClass
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| shape_partition.cc | 102 dimensions_(dimension_partition_counts_.size()),
106 for (int i = 0; i < dimensions_.size(); ++i) {
107 dimensions_[i] = shape_.layout().minor_to_major(
115 const int64 dim_size = shape_.dimensions(dimensions_[i]);
132 std::vector<std::pair<int64, int64>> partition(dimensions_.size());
143 shape_.dimensions(dimensions_[i]) - partition[i].first;
|
| shape_partition.h | 97 std::vector<int64> dimensions_; member in class:xla::cpu::ShapePartitionIterator
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mirror_pad_op.h | 124 dimensions_ = impl_.dimensions();
126 eigen_assert(padding_[dim].first + op.offset() <= dimensions_[dim]);
127 eigen_assert(padding_[dim].second + op.offset() <= dimensions_[dim]);
128 dimensions_[dim] += padding_[dim].first + padding_[dim].second;
137 output_strides_[i + 1] = output_strides_[i] * dimensions_[i];
144 output_strides_[i - 1] = output_strides_[i] * dimensions_[i];
150 return dimensions_;
218 (dimensions_[dim] - padding_[dim].second) * output_strides_[dim];
324 Dimensions dimensions_;
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorFixedSize.h | 26 template<typename Scalar_, typename Dimensions_, int Options_, typename IndexType>
27 class TensorFixedSize : public TensorBase<TensorFixedSize<Scalar_, Dimensions_, Options_, IndexType> >
30 typedef TensorFixedSize<Scalar_, Dimensions_, Options_, IndexType> Self;
31 typedef TensorBase<TensorFixedSize<Scalar_, Dimensions_, Options_, IndexType> > Base;
48 typedef Dimensions_ Dimensions;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_instruction.cc | 111 instruction->dimensions_.push_back(dimension);
467 instruction->dimensions_.assign(dimensions.begin(), dimensions.end());
551 instruction->dimensions_.push_back(dimension);
578 instruction->dimensions_.assign(dimensions_to_reduce.begin(),
670 instruction->dimensions_.assign(broadcast_dimensions.begin(),
762 instruction->dimensions_.assign(dimensions.begin(), dimensions.end());
[all...] |
| hlo_instruction.h | 1380 std::vector<int64> dimensions_; member in class:xla::HloInstruction
[all...] |
| /external/mesa3d/include/CL/ |
| cl.hpp | 2215 cl_uint dimensions_; member in class:cl::NDRange
[all...] |
| /external/tensorflow/tensorflow/docs_src/performance/xla/ |
| operation_semantics.md | [all...] |
