Home
last modified time | relevance | path

Searched refs:extended_output_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmul.h99 const RuntimeShape extended_output_shape = in BroadcastMul4DSlow() local
102 for (int b = 0; b < extended_output_shape.Dims(0); ++b) { in BroadcastMul4DSlow()
103 for (int y = 0; y < extended_output_shape.Dims(1); ++y) { in BroadcastMul4DSlow()
104 for (int x = 0; x < extended_output_shape.Dims(2); ++x) { in BroadcastMul4DSlow()
105 for (int c = 0; c < extended_output_shape.Dims(3); ++c) { in BroadcastMul4DSlow()
120 output_data[Offset(extended_output_shape, b, y, x, c)] = in BroadcastMul4DSlow()
Dadd.h89 const RuntimeShape extended_output_shape = in BroadcastAdd4DSlow() local
103 for (int b = 0; b < extended_output_shape.Dims(0); ++b) { in BroadcastAdd4DSlow()
104 for (int y = 0; y < extended_output_shape.Dims(1); ++y) { in BroadcastAdd4DSlow()
105 for (int x = 0; x < extended_output_shape.Dims(2); ++x) { in BroadcastAdd4DSlow()
106 for (int c = 0; c < extended_output_shape.Dims(3); ++c) { in BroadcastAdd4DSlow()
133 output_data[Offset(extended_output_shape, b, y, x, c)] = in BroadcastAdd4DSlow()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h705 const RuntimeShape extended_output_shape = in BroadcastAdd4DSlow() local
719 for (int b = 0; b < extended_output_shape.Dims(0); ++b) { in BroadcastAdd4DSlow()
720 for (int y = 0; y < extended_output_shape.Dims(1); ++y) { in BroadcastAdd4DSlow()
721 for (int x = 0; x < extended_output_shape.Dims(2); ++x) { in BroadcastAdd4DSlow()
722 for (int c = 0; c < extended_output_shape.Dims(3); ++c) { in BroadcastAdd4DSlow()
723 output_data[Offset(extended_output_shape, b, y, x, c)] = in BroadcastAdd4DSlow()
746 const RuntimeShape extended_output_shape = in BroadcastAdd4DSlow() local
760 for (int b = 0; b < extended_output_shape.Dims(0); ++b) { in BroadcastAdd4DSlow()
761 for (int y = 0; y < extended_output_shape.Dims(1); ++y) { in BroadcastAdd4DSlow()
762 for (int x = 0; x < extended_output_shape.Dims(2); ++x) { in BroadcastAdd4DSlow()
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_transpose.cc50 Shape extended_output_shape = output_shape; in Transpose() local
51 ExtendShape(&extended_output_shape, 4); in Transpose()
56 const std::vector<int>& extended_output_dims = extended_output_shape.dims(); in Transpose()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc2098 Shape extended_output_shape = output_shape; in ShuffleArrayTemplate() local
2099 ExtendShape(&extended_output_shape, 4); in ShuffleArrayTemplate()
2104 const std::vector<int>& extended_output_dims = extended_output_shape.dims(); in ShuffleArrayTemplate()