Searched refs:computation_count (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | computation_placer.cc | 41 proto->set_computation_count(computation_count()); in Serialize()
42 for (int computation = 0; computation < computation_count(); ++computation) { in Serialize()
54 TF_RET_CHECK(proto.computation_devices_size() == proto.computation_count()); in Deserialize()
55 if (proto.replica_count() <= 0 || proto.computation_count() <= 0) { in Deserialize()
59 proto.replica_count(), proto.computation_count()); in Deserialize()
62 proto.computation_count()); in Deserialize()
63 for (int computation = 0; computation < proto.computation_count(); in Deserialize()
78 int computation_count) { in DeviceId() argument
80 TF_RET_CHECK(computation < computation_count); in DeviceId()
86 int replica_count, int computation_count) { in AssignDevices() argument
[all …]
|
| D | computation_placer.h | 41 DeviceAssignment(int replica_count, int computation_count) in DeviceAssignment() argument
42 : Array2D<int>(replica_count, computation_count, -1) { in DeviceAssignment()
44 CHECK_GT(computation_count, 0); in DeviceAssignment()
48 int computation_count() const { return width(); } in computation_count() function
71 int replica_count, int computation_count);
76 int computation_count);
|
| D | hlo_subcomputation_unification_test.cc | 88 EXPECT_EQ(3, module->computation_count()); in TEST_F()
101 EXPECT_EQ(2, module->computation_count()); in TEST_F()
127 EXPECT_EQ(3, module->computation_count()); in TEST_F()
140 EXPECT_EQ(2, module->computation_count()); in TEST_F()
167 EXPECT_EQ(3, module->computation_count()); in TEST_F()
180 EXPECT_EQ(3, module->computation_count()); in TEST_F()
204 EXPECT_EQ(1, module->computation_count()); in TEST_F()
|
| D | hlo_execution_profile.h | 57 size_t computation_count() const { in computation_count() function
62 return instruction_count() + computation_count(); in total_count()
|
| D | flatten_call_graph_test.cc | 217 EXPECT_EQ(7, module->computation_count()); in TEST_F()
244 EXPECT_EQ(2, module->computation_count()); in TEST_F()
250 EXPECT_EQ(3, module->computation_count()); in TEST_F()
|
| D | hlo_cost_analysis.cc | 89 auto computation_count = ShapeUtil::ElementsIn(shape); in HandleElementwiseOp() local
96 current_properties_[kTranscendentalsKey] = computation_count; in HandleElementwiseOp()
100 current_properties_[kFlopsKey] = computation_count; in HandleElementwiseOp()
|
| D | hlo_execution_profile.cc | 55 hlo_profile_index_map.computation_count()); in CreateHloProfilePrinterData()
|
| D | hlo_module.h | 135 int64 computation_count() const { return computations_.size(); } in computation_count() function
|
| D | buffer_assignment.cc | 439 if (module_sequence.size() == module_->computation_count()) { in ComputeSummaryStats()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 313 int32 computation_count = 2; field
|