OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:shape_size_
(Results
1 - 5
of
5
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_parallelization_preparation.h
42
: max_parallelism_(max_parallelism),
shape_size_
(shape_size) {}
74
const HloCostAnalysis::ShapeSizeFunction
shape_size_
;
member in class:xla::cpu::ParallelizationPreparation
parallel_task_assignment.cc
32
: max_parallelism_(max_parallelism),
shape_size_
(shape_size) {}
37
const int64 instruction_cost =
shape_size_
(instruction->shape());
46
const HloCostAnalysis::ShapeSizeFunction
shape_size_
;
member in class:xla::cpu::SimpleCostModel
55
shape_size_
(shape_size),
78
instruction_cost =
shape_size_
(instruction->shape());
101
const HloCostAnalysis::ShapeSizeFunction
shape_size_
;
member in class:xla::cpu::DefaultCostModel
cpu_parallelization_preparation.cc
114
ParallelTaskAssignment parallel_task_assignment(max_parallelism_,
shape_size_
,
/external/tensorflow/tensorflow/compiler/xla/service/
hlo_cost_analysis.cc
39
:
shape_size_
(shape_size), per_second_rates_(per_second_rates) {}
51
float bytes_accessed =
shape_size_
(hlo->shape());
53
bytes_accessed +=
shape_size_
(operand->shape());
185
current_properties_[kBytesAccessedKey] =
shape_size_
(tuple->shape());
458
current_properties_[kBytesAccessedKey] +=
shape_size_
(subshape);
462
current_properties_[kBytesAccessedKey] +=
shape_size_
(operand->shape());
580
shape_size = &
shape_size_
;
hlo_cost_analysis.h
149
// given hlo. Uses
shape_size_
to calculate shape sizes if shape_size is null,
150
// otherwise uses
shape_size_
.
173
const ShapeSizeFunction
shape_size_
;
member in class:xla::HloCostAnalysis
Completed in 648 milliseconds