Searched refs:computation_count (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | computation_placer.cc | 43 proto->set_computation_count(computation_count()); in Serialize()
44 for (int computation = 0; computation < computation_count(); ++computation) { in Serialize()
56 TF_RET_CHECK(proto.computation_devices_size() == proto.computation_count()); in Deserialize()
57 if (proto.replica_count() <= 0 || proto.computation_count() <= 0) { in Deserialize()
61 proto.replica_count(), proto.computation_count()); in Deserialize()
64 proto.replica_count(), proto.computation_count()); in Deserialize()
65 for (int computation = 0; computation < proto.computation_count(); in Deserialize()
79 string output = StrCat("Computations: ", computation_count(), in ToString()
81 for (int computation = 0; computation < computation_count(); ++computation) { in ToString()
93 int computation_count) { in DeviceId() 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
73 int replica_count, int computation_count);
78 int computation_count);
|
| D | hlo_subcomputation_unification_test.cc | 87 EXPECT_EQ(3, module->computation_count()); in TEST_F()
90 EXPECT_EQ(2, module->computation_count()); in TEST_F()
116 EXPECT_EQ(3, module->computation_count()); in TEST_F()
119 EXPECT_EQ(2, module->computation_count()); in TEST_F()
146 EXPECT_EQ(3, module->computation_count()); in TEST_F()
149 EXPECT_EQ(3, module->computation_count()); in TEST_F()
173 EXPECT_EQ(1, module->computation_count()); in TEST_F()
|
| D | hlo_execution_profile.h | 65 size_t computation_count() const { in computation_count() function
74 return instruction_count() + computation_count() + extra_metrics_count(); in total_count()
|
| D | flatten_call_graph_test.cc | 218 EXPECT_EQ(7, module->computation_count()); in TEST_F()
245 EXPECT_EQ(2, module->computation_count()); in TEST_F()
251 EXPECT_EQ(3, module->computation_count()); in TEST_F()
|
| D | hlo_schedule_test.cc | 331 ASSERT_EQ(module->computation_count(), 3); in TEST_F()
333 ASSERT_EQ(module->computation_count(), 1); in TEST_F()
|
| D | hlo_execution_profile.cc | 62 hlo_profile_index_map.computation_count()); in CreateHloProfilePrinterData()
|
| D | hlo_cost_analysis.cc | 90 auto computation_count = ShapeUtil::ElementsIn(shape); in HandleElementwiseOp() local
98 current_properties_[kTranscendentalsKey] = computation_count; in HandleElementwiseOp()
102 current_properties_[kFlopsKey] = computation_count; in HandleElementwiseOp()
|
| D | hlo_module.h | 168 int64 computation_count() const { return computations_.size(); } in computation_count() function
|
| D | hlo_parser.cc | 4114 TF_RET_CHECK(module->computation_count() == 0); in ParseHloString()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.cc | 89 if (computation_count() == 0) { in Verify()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 344 int32 computation_count = 2; field
|