OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:perm_shape
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/lite/kernels/
transpose_test.cc
173
// SpaceToBatchNDOpConstModel m(input_shape,
perm_shape
, perm_data);
179
std::initializer_list<int>
perm_shape
,
182
perm_ = AddConstInput(TensorType_INT32, perm,
perm_shape
);
193
// TransposeOpDynamicModel m(input_shape,
perm_shape
);
200
std::initializer_list<int>
perm_shape
) {
206
BuildInterpreter({input_shape,
perm_shape
});
/external/tensorflow/tensorflow/lite/delegates/nnapi/
nnapi_delegate_test.cc
801
std::initializer_list<int>
perm_shape
,
804
perm_ = AddConstInput(TensorType_INT32, perm,
perm_shape
);
808
BuildInterpreter({input_shape,
perm_shape
});
[
all
...]
/external/tensorflow/tensorflow/core/ops/
array_ops.cc
116
ShapeHandle
perm_shape
= c->input(1);
local
118
DimensionHandle perm_elems = c->NumElements(
perm_shape
);
148
TF_RETURN_IF_ERROR(c->WithRank(
perm_shape
, 1, &
perm_shape
));
[
all
...]
Completed in 182 milliseconds