OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:computation_count
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/
computation_placer.h
41
DeviceAssignment(int replica_count, int
computation_count
)
42
: Array2D<int>(replica_count,
computation_count
, -1) {
44
CHECK_GT(
computation_count
, 0);
48
int
computation_count
() const { return width(); }
function in class:xla::DeviceAssignment
68
// instance for [replica_count x
computation_count
] setup. The returned device
71
int replica_count, int
computation_count
);
76
int
computation_count
);
hlo_execution_profile.h
57
size_t
computation_count
() const {
function in class:xla::HloProfileIndexMap
62
return instruction_count() +
computation_count
();
hlo_module.h
135
int64
computation_count
() const { return computations_.size(); }
function in class:xla::HloModule
hlo_cost_analysis.cc
89
auto
computation_count
= ShapeUtil::ElementsIn(shape);
local
96
current_properties_[kTranscendentalsKey] =
computation_count
;
100
current_properties_[kFlopsKey] =
computation_count
;
Completed in 272 milliseconds