Searched refs:dimensions_ (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape.h | 52 return dimensions_.size(); in rank()
96 int dimensions_size() const { return dimensions_.size(); } in dimensions_size()
97 int64 dimensions(int index) const { return dimensions_.at(index); } in dimensions()
98 void set_dimensions(int index, int64 value) { dimensions_.at(index) = value; } in set_dimensions()
100 dimensions_.push_back(value); in add_dimensions()
104 dimensions_.clear(); in clear_dimensions()
107 const std::vector<int64>& dimensions() const { return dimensions_; } in dimensions()
108 absl::Span<int64> mutable_dimensions() { return absl::MakeSpan(dimensions_); } in mutable_dimensions()
136 dimensions_.clear(); in Clear()
205 std::vector<int64> dimensions_; variable
|
| D | layout.h | 37 : dimensions_(dimensions.begin(), dimensions.end()) {} in Tile()
53 int64 dimension(int i) const { return dimensions_.at(i); } in dimension()
56 const std::vector<int64>& dimensions() const { return dimensions_; } in dimensions()
59 dimensions_.push_back(value); in add_dimensions()
64 dimensions_.clear(); in clear_dimensions()
74 std::vector<int64> 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/ |
| D | mirror_pad_op.h | 125 dimensions_ = impl_.dimensions();
127 eigen_assert(padding_[dim].first + op.offset() <= dimensions_[dim]);
128 eigen_assert(padding_[dim].second + op.offset() <= dimensions_[dim]);
129 dimensions_[dim] += padding_[dim].first + padding_[dim].second;
138 output_strides_[i + 1] = output_strides_[i] * dimensions_[i];
145 output_strides_[i - 1] = output_strides_[i] * dimensions_[i];
151 return dimensions_;
219 (dimensions_[dim] - padding_[dim].second) * output_strides_[dim];
325 Dimensions dimensions_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 404 const std::vector<int64>& dimensions() const override { return dimensions_; } in dimensions()
421 std::vector<int64> dimensions_; variable
430 const std::vector<int64>& dimensions() const override { return dimensions_; } in dimensions()
449 std::vector<int64> dimensions_; variable
459 const std::vector<int64>& dimensions() const override { return dimensions_; } in dimensions()
490 std::vector<int64> dimensions_; variable
499 const std::vector<int64>& dimensions() const override { return dimensions_; } in dimensions()
524 std::vector<int64> dimensions_; variable
533 const std::vector<int64>& dimensions() const override { return dimensions_; } in dimensions()
552 std::vector<int64> dimensions_; variable
[all …]
|
| D | hlo_instructions.cc | 655 dimensions_(dimensions.begin(), dimensions.end()) { in HloReverseInstruction()
661 for (int64 dimension : dimensions_) { in ToProto()
691 : HloInstruction(HloOpcode::kConcatenate, shape), dimensions_({dimension}) { in HloConcatenateInstruction()
699 for (int64 dimension : dimensions_) { in ToProto()
732 dimensions_(dimensions_to_reduce.begin(), dimensions_to_reduce.end()) { in HloReduceInstruction()
741 for (int64 dimension : dimensions_) { in ToProto()
776 dimensions_({dimension}), in HloSortInstruction()
786 for (int64 dimension : dimensions_) { in ToProto()
828 dimensions_(dimensions.begin(), dimensions.end()) { in HloTransposeInstruction()
841 for (int64 dimension : dimensions_) { in ToProto()
[all …]
|
| /external/mesa3d/include/CL/ |
| D | cl.hpp | 2215 cl_uint dimensions_; member in cl::NDRange
2219 : dimensions_(0) in NDRange()
2223 : dimensions_(1) in NDRange()
2229 : dimensions_(2) in NDRange()
2236 : dimensions_(3) in NDRange()
2244 ::size_t dimensions() const { return dimensions_; } in dimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2124 … | `ArraySlice<int64>` | The set of dimensions in `updates` shape that are _window dimensions_.
2125 `inserted_window_dims` | `ArraySlice<int64>` | The set of _window dimensions_ that must be …
|