OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:physical_dim
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_batchnorm_thunk.cc
39
auto physical_dim_size = [&](int64
physical_dim
) {
40
return shape.dimensions(LayoutUtil::Major(shape.layout(),
physical_dim
));
50
int64
physical_dim
;
local
51
for (
physical_dim
= 0;
physical_dim
!= logical_to_physical[feature_index];
52
++
physical_dim
) {
53
CHECK_LT(
physical_dim
, shape.dimensions_size());
54
batch_size *= physical_dim_size(
physical_dim
);
56
++
physical_dim
; // Skip the feature dimension.
57
for (;
physical_dim
< shape.dimensions_size(); ++physical_dim)
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ir_array.cc
274
int64
physical_dim
= logical_to_physical[dimension_mapping[i]];
local
276
std::min(min_broadcasted_dimension,
physical_dim
);
278
std::max(max_broadcasted_dimension,
physical_dim
);
Completed in 390 milliseconds