Searched refs:DynamicDimension (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_parameter_binding.h | 54 struct DynamicDimension { struct
64 friend H AbslHashValue(H h, const DynamicDimension& m) { in AbslHashValue() argument
69 friend bool operator==(const DynamicDimension& lhs,
70 const DynamicDimension& rhs) {
85 const DynamicDimension& dynamic_dimension);
92 const DynamicDimension& dynamic_dimension) const;
96 const DynamicDimension& dynamic_dimension)>;
117 absl::flat_hash_map<DynamicDimension, DynamicParameter> bindings_;
|
| D | dynamic_dimension_inference.h | 57 struct DynamicDimension { struct
70 friend H AbslHashValue(H h, const DynamicDimension& m) { in AbslHashValue() argument
74 friend bool operator==(const DynamicDimension& lhs,
75 const DynamicDimension& rhs) {
86 dynamic_mapping_.try_emplace(DynamicDimension{inst, index, dim}, size); in SetDynamicSize()
88 iter.first->second.emplace(DynamicDimension{inst, index, dim}); in SetDynamicSize()
106 using DynamicMapping = absl::flat_hash_map<DynamicDimension, HloInstruction*>;
113 absl::flat_hash_set<DynamicDimension>>;
|
| D | dynamic_parameter_binding_test.cc | 65 DynamicParameterBinding::DynamicDimension{1, {}, 0})); in TEST_F()
70 DynamicParameterBinding::DynamicDimension{/*parameter_num=*/1, in TEST_F()
103 DynamicParameterBinding::DynamicDimension{0, {1}, 0})); in TEST_F()
108 DynamicParameterBinding::DynamicDimension{/*parameter_num=*/0, in TEST_F()
142 DynamicParameterBinding::DynamicDimension{0, {1}, 0})); in TEST_F()
146 DynamicParameterBinding::DynamicDimension{0, {1}, 1})); in TEST_F()
151 DynamicParameterBinding::DynamicDimension{/*parameter_num=*/0, in TEST_F()
162 DynamicParameterBinding::DynamicDimension{/*parameter_num=*/0, in TEST_F()
|
| D | dynamic_dimension_inference_test.cc | 95 DynamicParameterBinding::DynamicDimension{0, {}, 1})); in TEST_F()
114 DynamicParameterBinding::DynamicDimension{0, {0}, 1})); in TEST_F()
140 DynamicParameterBinding::DynamicDimension{0, {0}, 1})); in TEST_F()
171 DynamicParameterBinding::DynamicDimension{0, {}, 1})); in TEST_F()
202 DynamicParameterBinding::DynamicDimension{0, {}, 1})); in TEST_F()
234 DynamicParameterBinding::DynamicDimension{0, {}, 2})); in TEST_F()
270 DynamicParameterBinding::DynamicDimension{0, {}, 0})); in TEST_F()
275 DynamicParameterBinding::DynamicDimension{0, {}, 1})); in TEST_F()
278 DynamicParameterBinding::DynamicDimension{1, {}, 0})); in TEST_F()
322 DynamicParameterBinding::DynamicDimension{0, {}, 0})); in TEST_F()
[all …]
|
| D | dynamic_parameter_binding.cc | 25 const DynamicDimension& dynamic_dimension) { in Bind()
33 const DynamicDimension& dynamic_dimension) const { in GetBinding()
44 const DynamicDimension& dynamic_dimension = binding.first; in ToProto()
78 DynamicDimension{target_param_num, target_param_index, in CreateFromProto()
89 const DynamicDimension& dynamic_dimension = binding.first; in ToString()
113 const DynamicDimension& dynamic_dimension) in Verify()
|
| D | dynamic_padder_test.cc | 95 DynamicParameterBinding::DynamicDimension{0, {}, 1})); in TEST_F()
138 DynamicParameterBinding::DynamicDimension{0, {}, 1})); in TEST_F()
181 DynamicParameterBinding::DynamicDimension{0, {}, 0})); in TEST_F()
208 DynamicParameterBinding::DynamicDimension{0, {}, 1})); in TEST_F()
|
| D | dynamic_dimension_inference.cc | 511 DynamicParameterBinding::DynamicDimension dynamic_dimension{ in HandleWhile()
531 const DynamicParameterBinding::DynamicDimension& dynamic_dimension) { in HandleParameter()
602 const DynamicDimension& dynamic_dimension = mapping.first; in ToString()
622 auto iter = dynamic_mapping_.find(DynamicDimension{inst, index, dim}); in GetDynamicSize()
|
| D | hlo_evaluator_test.cc | 3048 DynamicParameterBinding::DynamicDimension{1, {}, 0})); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 286 DynamicParameterBinding::DynamicDimension{ in SetDynamicBinding()
|