| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| batch_to_space_nd.cc | 20 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
131 TF_LITE_BATCH_TO_SPACE_ND(reference_ops, float);
138 TF_LITE_BATCH_TO_SPACE_ND(reference_ops, uint8_t);
145 TF_LITE_BATCH_TO_SPACE_ND(reference_ops, int32_t);
152 TF_LITE_BATCH_TO_SPACE_ND(reference_ops, int64_t);
|
| pad.cc | 20 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
131 TF_LITE_PAD(reference_ops, float);
138 TF_LITE_PAD(reference_ops, uint8_t);
145 TF_LITE_PAD(reference_ops, int32_t);
152 TF_LITE_PAD(reference_ops, int64_t);
|
| space_to_batch_nd.cc | 20 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
128 TF_LITE_SPACE_TO_BATCH_ND(reference_ops, float);
135 TF_LITE_SPACE_TO_BATCH_ND(reference_ops, uint8_t);
142 TF_LITE_SPACE_TO_BATCH_ND(reference_ops, int32_t);
149 TF_LITE_SPACE_TO_BATCH_ND(reference_ops, int64_t);
|
| space_to_depth.cc | 18 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
89 TF_LITE_SPACE_TO_DEPTH(reference_ops, float);
96 TF_LITE_SPACE_TO_DEPTH(reference_ops, uint8_t);
103 TF_LITE_SPACE_TO_DEPTH(reference_ops, int32_t);
110 TF_LITE_SPACE_TO_DEPTH(reference_ops, int64_t);
|
| transpose.cc | 19 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
119 TF_LITE_TRANSPOSE(reference_ops, float);
124 TF_LITE_TRANSPOSE(reference_ops, uint8_t);
129 TF_LITE_TRANSPOSE(reference_ops, int32_t);
134 TF_LITE_TRANSPOSE(reference_ops, int64_t);
|
| softmax_test.cc | 24 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
97 tflite::reference_ops::Softmax(input_buffer, input_dims, beta,
125 tflite::reference_ops::Softmax(input_buffer, input_dims, beta,
|
| exp.cc | 19 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
66 TF_LITE_EXP(reference_ops, float);
|
| mean.cc | 19 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
201 TF_LITE_MEAN(reference_ops, float);
204 TF_LITE_MEAN(reference_ops, int);
207 TF_LITE_MEAN(reference_ops, uint8_t);
210 TF_LITE_MEAN(reference_ops, int64_t);
|
| concatenation.cc | 26 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
121 TF_LITE_CONCATENATION(reference_ops, float);
128 TF_LITE_CONCATENATION(reference_ops, uint8_t);
|
| l2norm.cc | 18 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
72 TF_LITE_L2NORM(reference_ops);
|
| add.cc | 19 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
95 TF_LITE_ADD(reference_ops, BroadcastAdd);
97 TF_LITE_ADD(reference_ops, Add);
155 TF_LITE_ADD(reference_ops, BroadcastAdd);
|
| mul.cc | 19 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
95 TF_LITE_MUL(reference_ops, BroadcastMul);
97 TF_LITE_MUL(reference_ops, Mul);
140 TF_LITE_MUL(reference_ops, BroadcastMul);
|
| pooling.cc | 26 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
136 TF_LITE_AVERAGE_POOL(reference_ops);
159 TF_LITE_AVERAGE_POOL(reference_ops);
180 TF_LITE_MAX_POOL(reference_ops);
203 TF_LITE_MAX_POOL(reference_ops);
224 TF_LITE_L2_POOL(reference_ops);
|
| strided_slice.cc | 20 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
220 TF_LITE_STRIDED_SLICE(reference_ops, float);
225 TF_LITE_STRIDED_SLICE(reference_ops, int32_t);
230 TF_LITE_STRIDED_SLICE(reference_ops, int64_t);
|
| div.cc | 19 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
74 TF_LITE_DIV(reference_ops);
|
| local_response_norm.cc | 18 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
73 TF_LITE_LOCAL_RESPONSE_NORM(reference_ops);
|
| sub.cc | 19 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
74 TF_LITE_Sub(reference_ops);
|
| resize_bilinear.cc | 18 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
98 TF_LITE_RESIZE_BILINEAR(reference_ops);
|
| split.cc | 20 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
126 reference_ops::TensorFlowSplit<scalar>( \
|
| depthwise_conv.cc | 182 depthwise_conv = &reference_ops::DepthwiseConv;
210 depthwise_conv = &reference_ops::DepthwiseConv;
|
| /frameworks/ml/nn/common/operations/ |
| StridedSlice.cpp | 24 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
73 tflite::reference_ops::StridedSlice(
81 tflite::reference_ops::StridedSlice(
|
| FullyConnected.cpp | 21 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
45 tflite::reference_ops::FullyConnected(
|
| Reshape.cpp | 25 #include "tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h"
215 tflite::reference_ops::Transpose(
222 tflite::reference_ops::Transpose(
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
| depthwiseconv_float.h | 23 namespace reference_ops { namespace in namespace:tflite
112 } // end namespace reference_ops
|
| depthwiseconv_uint8.h | 27 namespace reference_ops { namespace in namespace:tflite
135 } // end namespace reference_ops
|
