| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 71 ShapeIndex(std::initializer_list<int64> init) : indices_(init) {} in ShapeIndex()
73 ShapeIndex(InputIt start, InputIt end) : indices_(start, end) {} in ShapeIndex()
77 bool empty() const { return indices_.empty(); } in empty()
78 size_t size() const { return indices_.size(); } in size()
79 void push_back(int64 value) { indices_.push_back(value); } in push_back()
80 void pop_back() { indices_.pop_back(); } in pop_back()
83 void push_front(int64 value) { indices_.insert(indices_.begin(), value); } in push_front()
87 container_type::const_iterator begin() const { return indices_.begin(); } in begin()
88 container_type::const_iterator end() const { return indices_.end(); } in end()
89 container_type::iterator begin() { return indices_.begin(); } in begin()
[all …]
|
| D | shape_util.cc | 86 return StrCat("{", absl::StrJoin(indices_, ","), "}"); in ToString()
90 return indices_ == other.indices_; in operator ==()
109 indices_.subspan(0, prefix.size()) == prefix.indices_; in StartsWith()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_reachability.h | 103 i.v = FindOrDie(indices_, GetKey(instruction)); in GetIndex()
144 return indices_.contains(GetKey(a)); in IsPresent()
226 return FindOrDie(indices_, GetKey(instruction)); in GetIndexInternal()
234 absl::flat_hash_map<uint64, int> 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()
68 indices_[GetKey(replacement)] = GetIndex(original).v; in Replace()
69 indices_.erase(GetKey(original)); in Replace()
|
| D | indexed_array_analysis.h | 181 Array* indices() const { return indices_; } in indices()
197 indices_(indices), in ScalarIndexedArray()
203 Array* indices_; variable
|
| /external/tensorflow/tensorflow/lite/kernels/perception/ |
| D | max_unpooling_2d_test.cc | 39 indices_ = AddInput(indices); in MaxUnpoolingOpModel()
49 BuildInterpreter({GetShape(input_), GetShape(indices_)}); in MaxUnpoolingOpModel()
56 PopulateTensor(indices_, data); in SetIndices()
65 int indices_; member in tflite::ops::custom::__anon2bd9014c0111::MaxUnpoolingOpModel
|
| D | max_pool_with_argmax_test.cc | 43 indices_ = AddOutput(indices); in MaxpoolingWithArgMaxOpModel()
59 std::vector<int32_t> GetIndices() { return ExtractVector<int32_t>(indices_); } in GetIndices()
61 std::vector<int> GetIndicesShape() { return GetTensorShape(indices_); } in GetIndicesShape()
66 int indices_; member in tflite::ops::custom::__anon354936f00111::MaxpoolingWithArgMaxOpModel
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse_test.cc | 44 indices_ = AddInput(TensorType_INT32); in EmbeddingLookupSparseOpModel()
61 PopulateTensor(indices_, indices_data); in SetInput()
86 int indices_; member in tflite::__anonc8d704f40111::EmbeddingLookupSparseOpModel
|
| D | scatter_nd_test.cc | 34 indices_ = AddInput(indices); in ScatterNdOpModel()
41 {GetShape(indices_), GetShape(updates_), GetShape(shape_)}); in ScatterNdOpModel()
46 PopulateTensor<T>(indices_, data); in SetIndices()
67 int indices_; member in tflite::__anonb54fb7560111::ScatterNdOpModel
|
| D | one_hot_test.cc | 38 indices_ = AddInput(indices_type); in OneHotOpModel()
54 PopulateTensor<TI>(indices_, data); in SetIndices()
64 int indices_; member in tflite::__anon3d7a871b0111::OneHotOpModel
|
| D | sparse_to_dense_test.cc | 37 indices_ = AddInput(tensor_index_type); in SparseToDenseOpModel()
51 int indices() { return indices_; } in indices()
59 int indices_; member in tflite::__anon13f7662d0111::SparseToDenseOpModel
|
| D | gather_nd_test.cc | 37 indices_ = AddInput(indices); in GatherNdOpModel()
41 BuildInterpreter({GetShape(params_), GetShape(indices_)}); in GatherNdOpModel()
51 PopulateTensor<T>(indices_, data); in SetPositions()
63 int indices_; member in tflite::__anon457cb77b0111::GatherNdOpModel
|
| /external/webrtc/rtc_tools/frame_analyzer/ |
| D | video_temporal_aligner.cc | 66 : video_(video), indices_(indices) {} in ReorderedVideo()
70 size_t number_of_frames() const override { return indices_.size(); } in number_of_frames()
74 return video_->GetFrame(indices_.at(index)); in GetFrame()
79 const std::vector<size_t> indices_; member in webrtc::test::__anond7f027040111::ReorderedVideo
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | one_hot_op.h | 43 : indices_(indices), on_value_(on_value), off_value_(off_value) {} in OneGenerator()
47 return (indices_(pre_depth_suff[0], pre_depth_suff[2]) == pre_depth_suff[1]) in operator()
53 const typename TTypes<TI>::ConstMatrix indices_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_execution_profiler.cc | 109 indices_.erase(index); in FinishHloInstruction()
122 CHECK(indices_.insert(*index).second) << *index; in MakeScopedInstructionProfiler()
|
| D | hlo_execution_profiler.h | 87 std::unordered_set<size_t> indices_; variable
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | sparse_tensor_slice_dataset_op_test.cc | 33 indices_(std::move(indices)),
41 return {indices_, values_, dense_shape_}; in GetInputTensors()
61 Tensor indices_; member in tensorflow::data::__anon116fed2d0111::SparseTensorSliceDatasetParams
|
| /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;
|