| /external/tensorflow/tensorflow/compiler/xla/service/ |
| liveness_util.h | 37 const ShapeIndex& index,
41 const ShapeIndex& index,
53 HloInstruction* operand, const ShapeIndex& operand_index,
54 HloInstruction* user, const ShapeIndex& user_index,
57 const ShapeIndex& operand_index,
59 const ShapeIndex& user_index,
|
| hlo_dataflow_analysis.h | 69 const ShapeIndex& index = {}) const;
76 const ShapeIndex& index = {}) const;
78 const ShapeIndex& index = {});
89 const ShapeIndex& index = {}) const;
93 const ShapeIndex& index = {});
98 const ShapeIndex& index = {}) const {
102 const ShapeIndex& index = {}) {
126 HloValue* NewHloValue(HloInstruction* instruction, const ShapeIndex& index,
|
| logical_buffer.h | 100 LogicalBuffer(HloInstruction* instruction, const ShapeIndex& index, Id id);
109 const ShapeIndex& index() const { return index_; }
153 const HloInstruction& instruction, const ShapeIndex& index);
163 ShapeIndex index_;
|
| tuple_points_to_analysis.h | 78 const ShapeIndex& index) const;
85 void AddPointedToBuffer(const LogicalBuffer& buffer, const ShapeIndex& index);
107 const SourceSet& tuple_sources(const ShapeIndex& index) const;
110 void add_tuple_source(const ShapeIndex& index, HloInstruction* tuple);
115 const BufferList& element(const ShapeIndex& index) const {
118 BufferList* mutable_element(const ShapeIndex& index) {
125 tree_.ForEachElement([&fn](const ShapeIndex& index, const Elem& elem) {
131 tree_.ForEachMutableElement([&fn](const ShapeIndex& index, Elem* elem) {
138 [&fn](const ShapeIndex& index, const Elem& elem) {
160 BufferAlias(HloInstruction* instruction, const ShapeIndex& index
[all...] |
| logical_buffer_analysis.h | 39 const ShapeIndex& index) const;
56 void NewLogicalBuffer(HloInstruction* instruction, const ShapeIndex& index);
72 std::pair<const HloInstruction*, const ShapeIndex> p) const {
83 std::unordered_map<std::pair<const HloInstruction*, const ShapeIndex>,
|
| hlo_alias_analysis.h | 65 const ShapeIndex& index = {}) const;
67 const ShapeIndex& index = {});
73 const HloInstruction* instruction, const ShapeIndex& index = {}) const;
|
| tuple_points_to_analysis.cc | 49 [&ambiguous](const ShapeIndex& /*index*/, const BufferList& points_to) {
58 ForEachElement([&distinct, &all_points_to](const ShapeIndex& /*index*/,
79 [&flat_set](const ShapeIndex& /*index*/, const BufferList& buffers) {
87 ForEachElement([&found, &buffer](const ShapeIndex& /*index*/,
99 const ShapeIndex& index) const {
106 const ShapeIndex& index) {
114 const ShapeIndex& index) const {
118 void PointsToSet::add_tuple_source(const ShapeIndex& index,
190 const ShapeIndex& index,
207 [this, hlo_instruction](const ShapeIndex& index
[all...] |
| logical_buffer.cc | 30 const ShapeIndex& index, Id id)
49 const HloInstruction& instruction, const ShapeIndex& index) {
|
| logical_buffer_analysis.cc | 84 const ShapeIndex& index) const {
89 const ShapeIndex& index) {
103 [this, hlo_instruction](const Shape& shape, const ShapeIndex& index) {
|
| hlo_value.h | 33 // HloPosition specifies a ShapeIndex within the output of a specific
37 ShapeIndex index;
68 ShapeIndex operand_index;
121 HloValue(Id id, HloInstruction* instruction, const ShapeIndex& index,
147 const ShapeIndex& defining_index() const { return defining_position().index; }
|
| liveness_util.cc | 32 const ShapeIndex& index,
73 const ShapeIndex& index,
110 HloInstruction* instruction, const ShapeIndex& index,
138 HloInstruction* operand, const ShapeIndex& operand_index,
184 HloInstruction* operand, const ShapeIndex& operand_index,
185 HloInstruction* user, const ShapeIndex& user_index,
276 const ShapeIndex& operand_index,
278 const ShapeIndex& user_index,
|
| shaped_buffer.h | 66 const ShapeIndex& index) const {
72 const ShapeIndex& index) {
|
| allocation_tracker.cc | 54 std::vector<ShapeIndex> shape_indices;
57 const ShapeIndex& index) {
60 for (const ShapeIndex& index : shape_indices) {
79 std::vector<ShapeIndex> shape_indices;
82 const ShapeIndex& index) {
85 for (const ShapeIndex& index : shape_indices) {
|
| hlo_dataflow_analysis.cc | 49 const ShapeIndex& index) const {
58 const HloInstruction* instruction, const ShapeIndex& index) const {
64 const HloInstruction* instruction, const ShapeIndex& index) {
70 const ShapeIndex& index,
126 const ShapeIndex& index,
168 const ShapeIndex& index = pair.first;
259 const HloInstruction* instruction, const ShapeIndex& index) const {
264 const ShapeIndex& index) {
311 const ShapeIndex& operand_index = pair.first;
314 ShapeIndex index = {0}
[all...] |
| buffer_assignment.h | 96 const ShapeIndex& param_shape_index() const {
214 ShapeIndex param_shape_index) {
249 ShapeIndex param_shape_index_;
295 const HloInstruction* instruction, const ShapeIndex& index) const;
300 const ShapeIndex& index) const;
310 const HloInstruction* instruction, const ShapeIndex& index) const;
323 const HloInstruction* instruction, const ShapeIndex& index) const {
332 const ShapeIndex& shape_index_a,
334 const ShapeIndex& shape_index_b) const;
|
| hlo_value.cc | 71 const ShapeIndex& index, bool is_phi)
113 // ShapeIndex in the given operand. Generally, instruction which pass through
116 bool MayUseOperandValue(int64 operand_number, const ShapeIndex& index,
262 const ShapeIndex& index = pair.first;
284 ForEachElement([this, &out](const ShapeIndex& index,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_hlo_support_checker.cc | 32 [&instruction](const Shape& subshape, const ShapeIndex&) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_hlo_support_checker.cc | 32 [&instruction](const Shape& subshape, const ShapeIndex&) {
|
| hlo_to_ir_bindings.h | 54 const ShapeIndex& shape_index = {});
72 // output of "inst".at the given ShapeIndex.
74 const ShapeIndex& shape_index = {}) const {
89 const ShapeIndex& shape_index = {});
100 const ShapeIndex& shape_index,
|
| hlo_to_ir_bindings.cc | 96 [&](const Shape& /*subshape*/, const ShapeIndex& index) {
140 const ShapeIndex& shape_index,
161 const ShapeIndex& shape_index) {
198 const ShapeIndex& shape_index) {
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_tree.h | 88 // duplicated. That is, every distinct ShapeIndex in the Shape has a unique T
140 const T& element(const ShapeIndex& index) const;
141 T* mutable_element(const ShapeIndex& index);
160 bool IsLeaf(const ShapeIndex& index) const {
165 // std::pair<ShapeIndex, T&>
249 // Fn : A callable of type void(const ShapeIndex& index, const T& data)
259 // void (const ShapeIndex& index, T* data).
271 // Copy the subtree of values from 'other' rooted at ShapeIndex
278 const ShapeIndex& source_base_index,
279 const ShapeIndex& target_base_index)
[all...] |
| shape_tree_test.cc | 63 int_tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) {
72 [&num_nodes](const ShapeIndex& /*index*/, bool data) {
89 tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) {
97 [&num_nodes](const ShapeIndex& /*index*/, int* data) {
105 tree.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) {
157 [&sum](const ShapeIndex& /*index*/, int data) { sum += data; });
169 [&sum](const ShapeIndex& /*index*/, int* data) { *data = 0; });
374 t.ForEachElement([&num_nodes](const ShapeIndex& /*index*/, int data) {
443 std::vector<ShapeIndex> v;
447 EXPECT_EQ(v, (std::vector<ShapeIndex>{{},
[all...] |
| shape_util.h | 39 // structures (trees) and ShapeIndex defines a path through the tree where each
40 // element of ShapeIndex indexes into a tuple (or nested tuple) within the
54 // ShapeIndex is a trivial wrapper around std::vector with a minimum number of
56 class ShapeIndex {
58 ShapeIndex() = default;
59 ShapeIndex(std::initializer_list<int64> init) : indices_(init) {}
82 bool operator==(const ShapeIndex& other) const {
85 bool operator!=(const ShapeIndex& other) const { return !(*this == other); }
86 bool operator<(const ShapeIndex& other) const {
96 // A view into a ShapeIndex as above, with the cheap/easy ability to consume th
[all...] |
| literal_util.h | 98 // given ShapeIndex is not array. See primitive_util.h for the mapping from
102 const ShapeIndex& shape_index = {}) const;
105 const ShapeIndex& shape_index = {});
110 const ShapeIndex& shape_index = {}) const;
111 SparseIndexArray* sparse_indices(const ShapeIndex& shape_index = {});
115 // given ShapeIndex is not array.
116 const void* untyped_data(const ShapeIndex& shape_index = {}) const;
117 void* untyped_data(const ShapeIndex& shape_index = {});
118 int64 size_bytes(const ShapeIndex& shape_index = {}) const;
238 const ShapeIndex& dest_shape_index = {}
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| ops.h | 51 ShapeIndex index;
|
