Searched refs:dimensions_ (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape.h | 46 dimensions_(dimensions.begin(), dimensions.end()), in Shape()
62 return dimensions_.size(); in rank()
112 int dimensions_size() const { return dimensions_.size(); } in dimensions_size()
113 int64 dimensions(int index) const { return dimensions_.at(index); } in dimensions()
114 void set_dimensions(int index, int64 value) { dimensions_.at(index) = value; } in set_dimensions()
116 dimensions_.push_back(value); in add_dimensions()
120 dimensions_.clear(); in clear_dimensions()
123 absl::Span<const int64> dimensions() const { return dimensions_; } in dimensions()
124 absl::Span<int64> mutable_dimensions() { return absl::MakeSpan(dimensions_); } in mutable_dimensions()
245 return H::combine(std::move(h), s.element_type_, s.dimensions_, in AbslHashValue()
[all …]
|
| D | layout.h | 37 : dimensions_(dimensions.begin(), dimensions.end()) {} in Tile()
53 int64 dimension(int i) const { return dimensions_.at(i); } in dimension()
56 absl::Span<const int64> dimensions() const { return dimensions_; } in dimensions()
59 dimensions_.push_back(value); in add_dimensions()
64 dimensions_.clear(); in clear_dimensions()
74 return H::combine(std::move(h), t.dimensions_); in AbslHashValue()
79 absl::InlinedVector<int64, 2> dimensions_;
|
| D | shape.cc | 26 dimensions_.reserve(shape_proto.dimensions_size()); in Shape()
100 CHECK_LT(dim_to_delete, dimensions_.size()); in DeleteDimension()
101 dimensions_.erase(dimensions_.begin() + dim_to_delete); in DeleteDimension()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | shape_partition.cc | 103 dimensions_(dimension_partition_counts_.size()), in ShapePartitionIterator()
107 for (int i = 0; i < dimensions_.size(); ++i) { in ShapePartitionIterator()
108 dimensions_[i] = shape_.layout().minor_to_major( in ShapePartitionIterator()
116 const int64 dim_size = shape_.dimensions(dimensions_[i]); in ShapePartitionIterator()
133 std::vector<std::pair<int64, int64>> partition(dimensions_.size()); in GetPartition()
144 shape_.dimensions(dimensions_[i]) - partition[i].first; in GetPartition()
|
| D | shape_partition.h | 97 std::vector<int64> dimensions_; variable
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | mirror_pad_op.h | 130 dimensions_ = impl_.dimensions();
132 eigen_assert(padding_[dim].first + op.offset() <= dimensions_[dim]);
133 eigen_assert(padding_[dim].second + op.offset() <= dimensions_[dim]);
134 dimensions_[dim] += padding_[dim].first + padding_[dim].second;
143 output_strides_[i + 1] = output_strides_[i] * dimensions_[i];
150 output_strides_[i - 1] = output_strides_[i] * dimensions_[i];
156 return dimensions_;
224 (dimensions_[dim] - padding_[dim].second) * output_strides_[dim];
226 const Index index_mod = index % (dimensions_[dim] * output_strides_[dim]);
331 Dimensions dimensions_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 535 const std::vector<int64>& dimensions() const override { return dimensions_; } in dimensions()
537 std::vector<int64>* mutable_dimensions() override { return &dimensions_; } in mutable_dimensions()
553 std::vector<int64> dimensions_; variable
562 const std::vector<int64>& dimensions() const override { return dimensions_; } in dimensions()
564 std::vector<int64>* mutable_dimensions() override { return &dimensions_; } in mutable_dimensions()
582 std::vector<int64> dimensions_; variable
592 const std::vector<int64>& dimensions() const override { return dimensions_; } in dimensions()
594 std::vector<int64>* mutable_dimensions() override { return &dimensions_; } in mutable_dimensions()
624 std::vector<int64> dimensions_; variable
633 const std::vector<int64>& dimensions() const override { return dimensions_; } in dimensions()
[all …]
|
| D | hlo_instructions.cc | 829 dimensions_(dimensions.begin(), dimensions.end()) { in HloReverseInstruction()
835 for (int64 dimension : dimensions_) { in ToProto()
865 : HloInstruction(HloOpcode::kConcatenate, shape), dimensions_({dimension}) { in HloConcatenateInstruction()
873 for (int64 dimension : dimensions_) { in ToProto()
906 dimensions_(dimensions_to_reduce.begin(), dimensions_to_reduce.end()) { in HloReduceInstruction()
915 for (int64 dimension : dimensions_) { in ToProto()
950 dimensions_({dimension}), in HloSortInstruction()
960 for (int64 dimension : dimensions_) { in ToProto()
1002 dimensions_(dimensions.begin(), dimensions.end()) { in HloTransposeInstruction()
1015 for (int64 dimension : dimensions_) { in ToProto()
[all …]
|
| /external/mesa3d/include/CL/ |
| D | cl2.hpp | 5692 cl_uint dimensions_; member in cl::NDRange
5697 : dimensions_(0) in NDRange()
5706 : dimensions_(1) in NDRange()
5715 : dimensions_(2) in NDRange()
5724 : dimensions_(3) in NDRange()
5742 return dimensions_; in dimensions()
5749 return dimensions_*sizeof(size_type); in size()
|
| D | cl.hpp | 4770 cl_uint dimensions_; member in cl::NDRange
4775 : dimensions_(0) in NDRange()
4780 : dimensions_(1) in NDRange()
4787 : dimensions_(2) in NDRange()
4795 : dimensions_(3) in NDRange()
4811 ::size_t dimensions() const { return dimensions_; } in dimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2402 … | `ArraySlice<int64>` | The set of dimensions in `updates` shape that are _window dimensions_.
2403 `inserted_window_dims` | `ArraySlice<int64>` | The set of _window dimensions_ that must be …
|