| /external/tensorflow/tensorflow/compiler/xla/ |
| D | sparse_index_array.cc | 28 : indices_(std::move(indices)), rank_(rank), max_indices_(max_indices) { in SparseIndexArray()
30 CHECK_EQ(indices_.size() % rank_, 0) in SparseIndexArray()
31 << "indices_.size(): " << indices_.size() << ", rank_: " << rank_; in SparseIndexArray()
47 CHECK_EQ(indices_.size() % rank_, 0); in index_count()
48 return indices_.size() / rank_; in index_count()
55 CHECK_LE(rank_ * sparse_element_number + rank_, indices_.size()); in At()
57 indices_.data() + rank_ * sparse_element_number, rank_); in At()
63 CHECK_LE(rank_ * sparse_element_number + rank_, indices_.size()); in At()
64 return absl::Span<int64>(indices_.data() + rank_ * sparse_element_number, in At()
71 indices_.insert(indices_.end(), index.begin(), index.end()); in Append()
[all …]
|
| D | shape_util.h | 68 ShapeIndex(std::initializer_list<int64> init) : indices_(init) {} in ShapeIndex()
70 ShapeIndex(InputIt start, InputIt end) : indices_(start, end) {} in ShapeIndex()
72 bool empty() const { return indices_.empty(); } in empty()
73 size_t size() const { return indices_.size(); } in size()
74 void push_back(int64 value) { indices_.push_back(value); } in push_back()
75 void pop_back() { indices_.pop_back(); } in pop_back()
78 void push_front(int64 value) { indices_.insert(indices_.begin(), value); } in push_front()
82 container_type::const_iterator begin() const { return indices_.begin(); } in begin()
83 container_type::const_iterator end() const { return indices_.end(); } in end()
84 container_type::iterator begin() { return indices_.begin(); } in begin()
[all …]
|
| D | sparse_index_array.h | 99 absl::Span<int64> mutable_data() { return absl::MakeSpan(indices_); } in mutable_data()
100 absl::Span<const int64> data() const { return indices_; } in data()
121 std::vector<int64> indices_;
|
| D | shape_util.cc | 57 return StrCat("{", absl::StrJoin(indices_, ","), "}"); in ToString()
61 return indices_ == other.indices_; in operator ==()
80 indices_.subspan(0, prefix.size()) == prefix.indices_; in StartsWith()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | one_hot_op.h | 37 : indices_(indices), on_value_(on_value), off_value_(off_value) {} in OneGenerator()
41 return (indices_(pre_depth_suff[0], pre_depth_suff[2]) == pre_depth_suff[1]) in operator()
47 const typename TTypes<TI>::ConstMatrix indices_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_reachability.h | 103 return indices_.contains(GetKey(a)); in IsPresent()
176 return FindOrDie(indices_, GetKey(instruction)); in GetIndex()
184 absl::flat_hash_map<uint64, int> indices_; variable
|
| D | indexed_array_analysis.h | 181 Array* indices() const { return indices_; } in indices()
197 indices_(indices), in ScalarIndexedArray()
203 Array* indices_; variable
|
| D | hlo_reachability.cc | 27 indices_[GetKey(hlo)] = bit_vectors_.size(); in HloReachabilityMap()
30 CHECK_EQ(size_, indices_.size()); // instructions should be unique in HloReachabilityMap()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse_test.cc | 40 indices_ = AddInput(TensorType_INT32); in EmbeddingLookupSparseOpModel()
57 PopulateTensor(indices_, indices_data); in SetInput()
81 int indices_; member in tflite::__anon2b9276160111::EmbeddingLookupSparseOpModel
|
| D | sparse_to_dense_test.cc | 36 indices_ = AddInput(tensor_index_type); in SparseToDenseOpModel()
50 int indices() { return indices_; } in indices()
58 int indices_; member in tflite::__anona42cd54f0111::SparseToDenseOpModel
|
| D | one_hot_test.cc | 35 indices_ = AddInput(indices_type); in OneHotOpModel()
51 PopulateTensor<TI>(indices_, data); in SetIndices()
61 int indices_; member in tflite::__anon580bf43d0111::OneHotOpModel
|
| D | gather_nd_test.cc | 32 indices_ = AddInput(indices); in GatherNdOpModel()
36 BuildInterpreter({GetShape(params_), GetShape(indices_)}); in GatherNdOpModel()
46 PopulateTensor<T>(indices_, data); in SetPositions()
58 int indices_; member in tflite::__anon4a1df51d0111::GatherNdOpModel
|
| /external/webrtc/webrtc/modules/video_render/ios/ |
| D | open_gles20.h | 59 static const char indices_[]; variable
|
| D | open_gles20.mm | 24 const char OpenGles20::indices_[] = {0, 3, 2, 0, 2, 1};
161 glDrawElements(GL_TRIANGLES, 6, GL_UNSIGNED_BYTE, indices_);
|
| /external/deqp/framework/opengl/ |
| D | gluDrawUtil.hpp | 174 PrimitiveList (PrimitiveType type_, int numElements_, IndexType indexType_, const void* indices_) in PrimitiveList()
178 , indices (indices_) in PrimitiveList()
|
| /external/eigen/unsupported/Eigen/CXX11/src/TensorSymmetry/ |
| D | StaticSymmetry.h | 31 template<typename indices_, int flags_>
34 typedef indices_ indices;
|