OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:dim_index
(Results
1 - 5
of
5
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
reduction_ops_common.cc
106
int
dim_index
= 0;
local
107
for (;
dim_index
< data.dims(); ++
dim_index
) {
108
if (data.dim_size(
dim_index
) != 1) break;
110
if (
dim_index
>= data.dims()) {
114
// Starting from the (
dim_index
)-th dimension, dimensions
124
reduce_first_axis_ = bitmap[
dim_index
];
125
data_reshape_.push_back(data.dim_size(
dim_index
));
126
++
dim_index
;
127
for (;
dim_index
< data.dims(); ++dim_index)
[
all
...]
/external/tensorflow/tensorflow/c/eager/
c_api_debug.cc
130
int64
dim_index
= padded_shape.layout().minor_to_major(i);
local
131
dev_dims.push_back(padded_shape.dimensions(
dim_index
));
159
TFE_TensorDebugInfo* debug_info, int
dim_index
) {
160
return debug_info->dev_dims[
dim_index
];
/external/skia/modules/skottie/src/
SkottieAdapter.cpp
306
const auto
dim_index
= SkTPin<size_t>(static_cast<size_t>(fDimensions),
local
313
fBlur->setSigma({ sigma * kDimensionsMap[
dim_index
].x(),
314
sigma * kDimensionsMap[
dim_index
].y() });
/external/tensorflow/tensorflow/core/framework/
shape_inference.cc
716
int64
dim_index
= dim_index_in;
local
717
if (
dim_index
< 0) {
718
dim_index
= s->dims_.size() +
dim_index
;
720
if (!FastBoundsCheck(
dim_index
, s->dims_.size())) {
722
return errors::InvalidArgument("Out of range
dim_index
", dim_index_in,
727
dims[
dim_index
] = new_dim;
[
all
...]
/external/tensorflow/tensorflow/core/grappler/costs/
graph_properties.cc
915
int
dim_index
member in struct:tensorflow::grappler::SymbolicShapeRefiner::DimId
[
all
...]
Completed in 188 milliseconds