Home
last modified time | relevance | path

Searched defs:output_shape (Results 1 – 25 of 90) sorted by relevance

1234

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/
A Dpad.h40 const P* pad_value_ptr, const RuntimeShape& output_shape, in PadImpl()
120 const P* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
130 const int32_t* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
141 const int32_t* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
151 const RuntimeShape& output_shape, T* output_data) { in PadImageStyle()
160 const RuntimeShape& output_shape, in PadImageStyle()
A Dsub.h37 const RuntimeShape& output_shape, in SubNonBroadcast()
53 const RuntimeShape& output_shape, in SubNonBroadcast()
74 const RuntimeShape& output_shape, in BroadcastSubSlow()
114 const RuntimeShape& output_shape, in BroadcastSubSlow()
171 const RuntimeShape& output_shape, in BroadcastSubSlow()
211 const RuntimeShape& output_shape, in BroadcastSubSlow()
302 const RuntimeShape& output_shape, T* output_data) { in BroadcastSubSlow()
341 const RuntimeShape& output_shape, in BroadcastSub16POTSlow()
449 const RuntimeShape& output_shape, uint8_t* output_data) { in Sub()
465 const RuntimeShape& output_shape, int8_t* output_data) { in Sub()
[all …]
A Dlogistic.h31 const RuntimeShape& output_shape, float* output_data) { in Logistic()
62 const float* input_data, const RuntimeShape& output_shape, in Logistic()
70 const RuntimeShape& output_shape, int16_t* output_data) { in Logistic()
95 const RuntimeShape& output_shape, int8_t* output_data, in Logistic()
A Dreference_ops.h93 const RuntimeShape& output_shape, T* output_data) { in Relu()
105 const RuntimeShape& output_shape, T* output_data) { in Relu1()
133 const RuntimeShape& output_shape, T* output_data) { in ReluX()
151 const RuntimeShape& output_shape, T* output_data) { in ReluX()
174 const RuntimeShape& output_shape, in BroadcastMulFivefold()
883 const RuntimeShape& output_shape, float* output_data) { in LocalResponseNormalization()
998 const RuntimeShape& output_shape, in GatherNdString()
1062 const RuntimeShape& output_shape, in Slice()
1312 const RuntimeShape output_shape = in SparseToDense() local
1348 const RuntimeShape& output_shape, T* output_data) { in Pow()
[all …]
A Dtanh.h30 const RuntimeShape& output_shape, float* output_data) { in Tanh()
43 const float* input_data, const RuntimeShape& output_shape, in Tanh()
50 const int16_t* input_data, const RuntimeShape& output_shape, in Tanh()
84 const uint8_t* input_data, const RuntimeShape& output_shape, in Tanh()
A Dmul.h48 const RuntimeShape& output_shape, T* output_data) { in Mul()
65 const RuntimeShape& output_shape, uint8_t* output_data) { in Mul()
79 const RuntimeShape& output_shape, in BroadcastMul4DSlow()
129 const RuntimeShape output_shape = in BroadcastMul4DSlow() local
A Dleaky_relu.h28 const RuntimeShape& output_shape, float* output_data) { in LeakyRelu()
41 const RuntimeShape& output_shape, in QuantizeLeakyRelu()
A Ddequantize.h34 const RuntimeShape& output_shape, OutputT* output_data) { in Dequantize()
51 const RuntimeShape& output_shape, float* output_data) { in PerChannelDequantize()
A Darg_min_max.h37 const T3* input2_data, const RuntimeShape& output_shape, in ArgMinMax()
79 const T3* input2_data, const RuntimeShape& output_shape, in ArgMinMax()
A Dlog_softmax.h30 const RuntimeShape& output_shape, float* output_data) { in LogSoftmax()
63 const RuntimeShape& output_shape, uint8_t* output_data) { in LogSoftmax()
158 const RuntimeShape& output_shape, in LogSoftmaxQuantized()
248 const RuntimeShape& output_shape, int8_t* output_data) { in LogSoftmax()
A Dhard_swish.h46 const RuntimeShape& output_shape, T* output_data) { in HardSwish()
61 const RuntimeShape& output_shape, T* output_data) { in HardSwish()
A Dbinary_function.h38 const RuntimeShape output_shape = in BroadcastBinaryFunction4DSlow() local
68 const RuntimeShape& output_shape, R* output_data, in BinaryFunction()
A Dcomparisons.h60 const T* input2_data, const RuntimeShape& output_shape, bool* output_data) { in ComparisonImpl()
74 const RuntimeShape& output_shape, bool* output_data) { in Comparison()
83 const T* input2_data, const RuntimeShape& output_shape, bool* output_data) { in ComparisonWithScaling()
110 const RuntimeShape output_shape; member
160 const RuntimeShape& output_shape, in BroadcastComparison4DSlow()
A Dadd.h31 const RuntimeShape& output_shape, T* output_data) { in Add()
118 const RuntimeShape& output_shape, uint8_t* output_data) { in Add()
136 const RuntimeShape& output_shape, in AddGeneralParamScale()
246 const RuntimeShape& output_shape, T* output_data) { in BroadcastAdd4DSlow()
308 const RuntimeShape& output_shape, in BroadcastAddFivefold()
A Dprelu.h31 const RuntimeShape& output_shape, T* output_data) { in BroadcastPrelu4DSlow()
78 const T* alpha_data, const RuntimeShape& output_shape, in Prelu()
A Dneg.h26 const RuntimeShape& output_shape, T* output_data) { in Negate()
A Dceil.h27 const RuntimeShape& output_shape, float* output_data) { in Ceil()
A Dfill.h27 const RuntimeShape& output_shape, T* output_data) { in Fill()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Dcomparisons.cc50 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in EqualEval() local
155 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in NotEqualEval() local
259 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in GreaterEval() local
350 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in GreaterEqualEval() local
441 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in LessEval() local
532 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in LessEqualEval() local
A Dactivations.cc52 const RuntimeShape& output_shape, const T* input_data, in ReluQuantized()
77 const RuntimeShape output_shape = GetTensorShape(output); in CalculateReluOpData() local
98 const RuntimeShape& output_shape, float* output_data) { in ReluFloat()
109 const RuntimeShape& output_shape, float* output_data) { in Relu6Float()
123 const RuntimeShape& output_shape, Q* output_data) { in Relu6Quantized()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/integer_ops/
A Dmul.h48 const RuntimeShape& output_shape, T* output_data) { in Mul()
62 const RuntimeShape& output_shape, int8_t* output_data) { in Mul()
92 const T* input2_data, const RuntimeShape& output_shape, T* output_data) { in BroadcastMul4DSlow()
A Dadd.h51 const int8_t* input2_data, const RuntimeShape& output_shape, in BroadcastBinaryFunction4DSlow()
120 const RuntimeShape& output_shape, int8_t* output_data) { in Add()
134 const RuntimeShape& output_shape, in BroadcastAdd4DSlow()
A Dtanh.h29 const RuntimeShape& output_shape, int8_t* output_data) { in Tanh()
64 const RuntimeShape& output_shape, int16_t* ptr_output_data) { in Tanh()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/
A Doptimized_ops.h370 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
1120 const RuntimeShape& output_shape; member
1139 const RuntimeShape output_shape = in Mean() local
1428 const RuntimeShape& output_shape, float* output_data, in HybridConvPerChannel()
1650 const RuntimeShape output_shape = in DepthToSpace() local
1691 const RuntimeShape output_shape = in SpaceToDepth() local
3695 const RuntimeShape& output_shape; member
4942 const RuntimeShape& output_shape, in Slice()
5012 const RuntimeShape output_shape = in StridedSlice() local
5595 const RuntimeShape output_shape = in ResizeNearestNeighbor() local
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/sparse_ops/
A Dfully_connected.h35 const RuntimeShape& output_shape, float* output_data) { in FullyConnectedSparseWeight()
81 const RuntimeShape& output_shape, float* output_data, int thread_start, in FullyConnectedSparseWeight1x4Impl()
153 const RuntimeShape& output_shape; member
169 const RuntimeShape& output_shape, float* output_data, in FullyConnectedSparseWeight1x4()

Completed in 49 milliseconds

1234