HomeSort by relevance Sort by last modified time
    Searched defs:DynamicDimension (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
dynamic_dimension_inference.h 54 // DynamicDimension is used as a key in the dynamic key-value mapping. It
57 struct DynamicDimension {
70 friend H AbslHashValue(H h, const DynamicDimension& m) {
74 friend bool operator==(const DynamicDimension& lhs,
75 const DynamicDimension& rhs) {
86 dynamic_mapping_.try_emplace(DynamicDimension{inst, index, dim}, size);
88 iter.first->second.emplace(DynamicDimension{inst, index, dim});
106 using DynamicMapping = absl::flat_hash_map<DynamicDimension, HloInstruction*>;
113 absl::flat_hash_set<DynamicDimension>>;
dynamic_parameter_binding.h 51 // DynamicDimension represents a dimension whose size is determined at
52 // runtime. A DynamicDimension's runtime size is determined by the binded
54 struct DynamicDimension {
64 friend H AbslHashValue(H h, const DynamicDimension& m) {
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)>;
113 // Keeps track of mappings from DynamicDimension to DynamicParameter. Th
    [all...]

Completed in 76 milliseconds