| /external/tensorflow/tensorflow/compiler/xla/ |
| sparse_index_array.cc | 28 : indices_(std::move(indices)), rank_(rank), max_indices_(max_indices) {
30 CHECK_EQ(indices_.size() % rank_, 0)
31 << "indices_.size(): " << indices_.size() << ", rank_: " << rank_;
47 CHECK_EQ(indices_.size() % rank_, 0);
48 return indices_.size() / rank_;
55 CHECK_LE(rank_ * sparse_element_number + rank_, indices_.size());
57 indices_.data() + rank_ * sparse_element_number, rank_);
64 CHECK_LE(rank_ * sparse_element_number + rank_, indices_.size());
66 indices_.data() + rank_ * sparse_element_number, rank_)
[all...] |
| shape_util.h | 59 ShapeIndex(std::initializer_list<int64> init) : indices_(init) {}
61 bool empty() const { return indices_.empty(); }
62 size_t size() const { return indices_.size(); }
63 void push_back(int64 value) { indices_.push_back(value); }
64 void pop_back() { indices_.pop_back(); }
67 void push_front(int64 value) { indices_.insert(indices_.begin(), value); }
69 std::vector<int64>::const_iterator begin() const { return indices_.begin(); }
70 std::vector<int64>::const_iterator end() const { return indices_.end(); }
71 std::vector<int64>::iterator begin() { return indices_.begin();
93 std::vector<int64> indices_; member in class:xla::ShapeIndex
[all...] |
| sparse_index_array.h | 98 tensorflow::gtl::MutableArraySlice<int64> mutable_data() { return &indices_; }
99 tensorflow::gtl::ArraySlice<int64> data() const { return indices_; }
120 std::vector<int64> indices_; member in class:xla::SparseIndexArray
|
| shape_util.cc | 46 "{", tensorflow::str_util::Join(indices_, ","), "}");
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| one_hot_op.h | 37 : indices_(indices), on_value_(on_value), off_value_(off_value) {}
41 return (indices_(pre_depth_suff[0], pre_depth_suff[2]) == pre_depth_suff[1])
47 const typename TTypes<TI>::ConstMatrix indices_; member in class:tensorflow::generator::OneGenerator
|
| /external/webrtc/webrtc/modules/video_render/ios/ |
| open_gles20.h | 59 static const char indices_[]; member in class:webrtc::OpenGles20
|
| open_gles20.mm | 24 const char OpenGles20::indices_[] = {0, 3, 2, 0, 2, 1};
161 glDrawElements(GL_TRIANGLES, 6, GL_UNSIGNED_BYTE, indices_);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_reachability.cc | 25 indices_[hlo] = bit_vectors_.size();
28 CHECK_EQ(size_, indices_.size()); // instructions should be unique
|
| hlo_reachability.h | 139 return FindOrDie(indices_, instruction);
147 tensorflow::gtl::FlatMap<const HloInstruction*, int> indices_; member in class:xla::HloReachabilityMap
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| embedding_lookup_sparse_test.cc | 40 indices_ = AddInput(TensorType_INT32);
57 PopulateTensor(indices_, indices_data);
81 int indices_; member in class:tflite::__anon39251::EmbeddingLookupSparseOpModel
|
| /external/deqp/framework/opengl/ |
| gluDrawUtil.hpp | 174 PrimitiveList (PrimitiveType type_, int numElements_, IndexType indexType_, const void* indices_)
178 , indices (indices_)
|
| /external/eigen/unsupported/Eigen/CXX11/src/TensorSymmetry/ |
| StaticSymmetry.h | 31 template<typename indices_, int flags_>
34 typedef indices_ indices;
|
| /toolchain/binutils/binutils-2.27/include/opcode/ |
| cgen.h | [all...] |
