HomeSort by relevance Sort by last modified time
    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