OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:AxesOrder
(Results
1 - 7
of
7
) sorted by null
/external/tensorflow/tensorflow/contrib/lite/toco/
tooling_util.h
271
int AxesCount(
AxesOrder
axes_order);
275
void GetShuffleShape(
AxesOrder
input_axes_order,
AxesOrder
output_axes_order,
283
void ShuffleDims(const Shape& input_shape,
AxesOrder
input_axes_order,
284
AxesOrder
output_axes_order, Shape* output_shape);
285
void ShuffleArray(const Shape& input_shape,
AxesOrder
input_axes_order,
286
AxesOrder
output_axes_order, const Shape& output_shape,
288
void ShuffleArray(const Shape& input_shape,
AxesOrder
input_axes_order,
289
AxesOrder
output_axes_order, const Shape& output_shape,
tooling_util.cc
[
all
...]
export_tensorflow.cc
108
void ExportFloatArray(
AxesOrder
input_axes_order, const Shape& input_shape,
109
const float* input_data,
AxesOrder
output_axes_order,
140
AxesOrder
input_axes_order,
141
AxesOrder
output_axes_order,
158
AxesOrder
input_axes_order,
159
AxesOrder
output_axes_order,
174
AxesOrder
input_axes_order,
175
AxesOrder
output_axes_order,
347
ConvertFloatTensorConst(model, weights_array_name,
AxesOrder
::kOHWI,
348
AxesOrder
::kHWIO, tensorflow_graph)
[
all
...]
model.h
122
// see
AxesOrder
, and this cannot always be resolved at the time of importing
135
enum class
AxesOrder
{
624
AxesOrder
input_axes_order;
625
AxesOrder
output_axes_order;
[
all
...]
import_tensorflow.cc
389
reorder->input_axes_order =
AxesOrder
::kHWIO;
390
reorder->output_axes_order =
AxesOrder
::kOHWI;
442
reorder->input_axes_order =
AxesOrder
::kHWIM;
443
reorder->output_axes_order =
AxesOrder
::k1HWO;
[
all
...]
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
convert_reorder_axes.cc
62
const
AxesOrder
& input_axes_order, const
AxesOrder
& output_axes_order) {
124
if (input_axes_order ==
AxesOrder
::kHWIM &&
125
output_axes_order ==
AxesOrder
::k1HWO) {
resolve_reorder_axes.cc
33
void ReorderAxes(
AxesOrder
input_axes_order,
AxesOrder
output_axes_order,
Completed in 97 milliseconds