| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_input_output_alias_config.h | 47 Alias(AliasKind kind, int64 parameter_number, ShapeIndex parameter_index)
54 ShapeIndex parameter_index;
66 Status SetUpAlias(const ShapeIndex& output_index, int64 param_number,
67 const ShapeIndex& param_index, AliasKind kind);
72 const ShapeIndex& param_index) const;
77 const ShapeIndex& param_index) const {
82 bool OutputHasAlias(const ShapeIndex& output_index) const;
94 absl::optional<ShapeIndex> GetAliasedOutput(
95 int64 param_number, const ShapeIndex& param_index) const;
101 const ShapeIndex& output_index) const
[all...] |
| logical_buffer.h | 37 LogicalBuffer(HloInstruction* instruction, const ShapeIndex& index, Id id);
45 const ShapeIndex& index() const override { return index_; }
58 ShapeIndex index_;
|
| optimize_input_output_buffer_alias_test.cc | 53 [&](const ShapeIndex&, const HloInputOutputAliasConfig::Alias&) {
94 EXPECT_EQ(config_.GetAliasedOutput(0, {0}), ShapeIndex{0});
95 EXPECT_EQ(config_.GetAliasedOutput(0, {1}), ShapeIndex{1});
96 EXPECT_EQ(config_.GetAliasedOutput(0, {2}), ShapeIndex{2});
97 EXPECT_EQ(config_.GetAliasedOutput(0, {3}), ShapeIndex{3});
110 EXPECT_EQ(config_.GetAliasedOutput(0, {0}), ShapeIndex{0});
111 EXPECT_EQ(config_.GetAliasedOutput(0, {2}), ShapeIndex{1});
124 EXPECT_EQ(config_.GetAliasedOutput(0, {0}), ShapeIndex{3});
125 EXPECT_EQ(config_.GetAliasedOutput(0, {1}), ShapeIndex{2});
126 EXPECT_EQ(config_.GetAliasedOutput(0, {2}), ShapeIndex{1})
[all...] |
| 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...] |
| hlo_dataflow_analysis.h | 86 const ShapeIndex& index = {}) const;
93 const ShapeIndex& index = {}) const;
95 const ShapeIndex& index = {});
106 const ShapeIndex& index = {}) const;
110 const ShapeIndex& index = {});
115 const ShapeIndex& index = {}) const {
119 const ShapeIndex& index = {}) {
144 const ShapeIndex& index,
152 const ShapeIndex& operand_index,
154 const ShapeIndex& user_index) const
[all...] |
| hlo_input_output_alias_config.cc | 22 const ShapeIndex& output_index) const {
26 Status HloInputOutputAliasConfig::SetUpAlias(const ShapeIndex& output_index,
28 const ShapeIndex& param_index,
57 [&](const ShapeIndex& index, const absl::optional<Alias>& data) {
88 ShapeIndex output_index(entry.output_shape_index().begin(),
91 ShapeIndex param_index(entry.parameter_shape_index().begin(),
109 ForEachAlias([&](const ShapeIndex& output_index, const Alias& alias) {
121 int64 param_number, const ShapeIndex& param_index) const {
124 [&](const xla::ShapeIndex&, absl::optional<Alias> alias) {
133 absl::optional<ShapeIndex> HloInputOutputAliasConfig::GetAliasedOutput
[all...] |
| bfloat16_propagation.h | 167 const ShapeIndex& index) const;
172 const ShapeIndex& index) const;
183 const ShapeIndex& index,
200 absl::flat_hash_map<HloInstruction*, absl::flat_hash_map<Shape*, ShapeIndex>>
|
| optimize_input_output_buffer_alias.cc | 50 absl::flat_hash_map<int64, std::queue<ShapeIndex>> size_to_input_index;
52 input_shape, [&](const Shape& subshape, const ShapeIndex& index) {
63 output_shape, [&](const Shape& subshape, const ShapeIndex& index) {
72 const ShapeIndex& input_index = it->second.front();
73 const ShapeIndex& output_index = index;
|
| hlo_liveness_analysis.h | 32 // Analysis which identifies all live {HloInstruction, ShapeIndex} pairs in
45 // which exports liveness for each {HloInstruction, ShapeIndex} in 'module'.
52 const ShapeIndex& shape_index) const;
|
| logical_buffer.cc | 27 const ShapeIndex& index, Id id)
|
| logical_buffer_analysis.h | 39 const ShapeIndex& index) const;
56 void NewLogicalBuffer(HloInstruction* instruction, const ShapeIndex& index);
74 std::pair<const HloInstruction*, const ShapeIndex> p) const {
85 std::unordered_map<std::pair<const HloInstruction*, const ShapeIndex>,
|
| buffer_value.h | 108 virtual const ShapeIndex& index() const = 0;
155 const HloInstruction& instruction, const ShapeIndex& index);
160 BufferValue(HloInstruction* instruction, const ShapeIndex& index, Id id);
|
| dynamic_dimension_inference.h | 46 HloInstruction* GetDynamicSize(HloInstruction* inst, const ShapeIndex& index,
61 ShapeIndex index;
84 void SetDynamicSize(HloInstruction* inst, const ShapeIndex& index, int64 dim,
|
| hlo_alias_analysis.h | 69 const ShapeIndex& index = {}) const;
71 const ShapeIndex& index = {});
77 const HloInstruction* instruction, const ShapeIndex& index = {}) const;
|
| hlo_input_output_alias_config_test.cc | 38 void expect_aliased(const ShapeIndex& output_index, int64 param_number,
39 const ShapeIndex& param_index,
41 absl::optional<ShapeIndex> aliased_output =
55 void expect_not_aliased(const ShapeIndex& output_index, int64 param_number,
56 const ShapeIndex& param_index,
58 absl::optional<ShapeIndex> aliased_output =
|
| dynamic_parameter_binding_test.cc | 75 EXPECT_EQ(param->parameter_index, ShapeIndex({}));
114 EXPECT_EQ(param->parameter_index, ShapeIndex({0}));
157 EXPECT_EQ(param->parameter_index, ShapeIndex({0}));
167 EXPECT_EQ(param2->parameter_index, ShapeIndex({0}));
|
| tuple_points_to_analysis.cc | 51 [&ambiguous](const ShapeIndex& /*index*/, const BufferList& points_to) {
60 ForEachElement([&](const ShapeIndex& /*index*/, const BufferList& points_to) {
80 [&flat_set](const ShapeIndex& /*index*/, const BufferList& buffers) {
88 ForEachElement([&found, &buffer](const ShapeIndex& /*index*/,
98 const ShapeIndex& index) const {
104 const ShapeIndex& index) {
112 const ShapeIndex& index) const {
116 void PointsToSet::add_tuple_source(const ShapeIndex& index,
187 const ShapeIndex& index,
204 [this, hlo_instruction](const ShapeIndex& index
[all...] |
| hlo_liveness_analysis.cc | 51 using VisitorFunction = std::function<void(const ShapeIndex& /*index*/)>;
55 index_tree.ForEachElement([&](const ShapeIndex& shape_index, bool live) {
67 const ShapeIndex& shape_index,
99 [&](const Shape& sub_shape, const ShapeIndex& shape_index) {
126 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index) {
134 ShapeIndex operand_shape_index;
157 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index) {
158 ShapeIndex operand_shape_index(shape_index);
176 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index) {
207 ForEachLiveIndex(index_tree, [&](const ShapeIndex& shape_index)
[all...] |
| buffer_value.cc | 28 BufferValue::BufferValue(HloInstruction* instruction, const ShapeIndex& index,
44 const HloInstruction& instruction, const ShapeIndex& index) {
|
| dynamic_parameter_binding.h | 48 ShapeIndex parameter_index;
58 ShapeIndex parameter_index;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_hlo_support_checker.cc | 31 [&instruction](const Shape& subshape, const ShapeIndex&) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_hlo_support_checker.cc | 31 [&instruction](const Shape& subshape, const ShapeIndex&) {
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| xrt_state.h | 106 const xla::ShapeIndex& subshape,
173 void DiscardAllocation(const xla::ShapeIndex& buffer_index);
182 const xla::ShapeIndex& source_index,
183 const xla::ShapeIndex& dest_index);
190 const std::function<bool(const xla::ShapeIndex&)>& release_checker);
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| literal.h | 75 // ShapeIndex is not array. See primitive_util.h for the mapping from XLA type
78 absl::Span<const NativeT> data(const ShapeIndex& shape_index = {}) const;
83 const ShapeIndex& shape_index = {}) const;
87 // the given ShapeIndex is not array.
88 const void* untyped_data(const ShapeIndex& shape_index = {}) const;
89 int64 size_bytes(const ShapeIndex& shape_index = {}) const;
114 const ShapeIndex& shape_index) const;
128 const ShapeIndex& shape_index = {}) const;
132 const ShapeIndex& shape_index = {}) const;
142 int64 sparse_element_number, const ShapeIndex& shape_index = {}) const
[all...] |
| shape_tree.h | 45 std::pair<ShapeIndex, T> data;
49 explicit ShapeTreeNode(ShapeIndex index)
51 ShapeTreeNode(ShapeIndex index, T data)
85 // duplicated. That is, every distinct ShapeIndex in the Shape has a unique T
147 // value_type = std::pair<ShapeIndex, T>.
152 std::pair<ShapeIndex, T>>;
156 const std::pair<ShapeIndex, T>>;
223 // Returns an iterator pointing to the given ShapeIndex.
242 // Fn : A callable of type void(const ShapeIndex& index, const T& data)
252 // void (const ShapeIndex& index, T* data)
[all...] |
