Searched refs:DummyTensor (Results 1 – 5 of 5) sorted by relevance
27 class DummyTensor(object): class41 a = DummyTensor(shape=(None, 32))42 b = DummyTensor(shape=(None, 32))59 a_2 = DummyTensor()61 b_2 = DummyTensor()84 a = DummyTensor()85 b = DummyTensor()88 a_2 = DummyTensor()90 b_2 = DummyTensor()94 merged = DummyTensor()[all …]
151 struct DummyTensor { struct155 bool operator==(const DummyTensor& b) const { argument163 ValueId Add(const DummyTensor& dummy) { in Add()167 void Add(ValueId id, const DummyTensor& dummy) { in Add()171 DummyTensor Get(ValueId id) { return reservations_[id]; } in Get()189 DummyTensor dummy; in Add()200 absl::flat_hash_map<ValueId, DummyTensor> reservations_;
553 std::vector<TensorUsageRecord<DummyTensor>> usage_records; in AllocateMemoryForStrongShapes()562 ObjectsAssignment<DummyTensor> assignment; in AllocateMemoryForStrongShapes()
166 struct DummyTensor { struct170 bool operator==(const DummyTensor& b) const { argument178 ValueId Add(const DummyTensor& dummy) { in Add()182 void Add(ValueId id, const DummyTensor& dummy) { in Add()186 DummyTensor Get(ValueId id) { return reservations_[id]; } in Get()204 DummyTensor dummy; in Add()215 absl::flat_hash_map<ValueId, DummyTensor> reservations_;
583 std::vector<TensorUsageRecord<DummyTensor>> usage_records; in AllocateMemoryForStrongShapes()592 ObjectsAssignment<DummyTensor> assignment; in AllocateMemoryForStrongShapes()