Home
last modified time | relevance | path

Searched defs:input_dims (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/common/operations/internal/reference/
Dreference_ops.h160 void Conv(const float* input_data, const Dims<4>& input_dims, in Conv()
219 void Conv(const uint8* input_data, const Dims<4>& input_dims, in Conv()
295 inline void DepthToSpace(const T* input_data, const Dims<4>& input_dims, in DepthToSpace()
336 inline void SpaceToDepth(const T* input_data, const Dims<4>& input_dims, in SpaceToDepth()
377 void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected()
410 void FullyConnected(const uint8* input_data, const Dims<4>& input_dims, in FullyConnected()
462 const float* input_data, const Dims<4>& input_dims, const float* mean_data, in NonGlobalBatchNormalization()
495 const Dims<4>& input_dims, const float* mean_data, in GlobalBatchNormalization()
524 inline void Relu(const float* input_data, const Dims<4>& input_dims, in Relu()
544 inline void Relu1(const float* input_data, const Dims<4>& input_dims, in Relu1()
[all …]
Ddepthwiseconv_float.h28 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv()
Ddepthwiseconv_uint8.h30 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv()
/frameworks/ml/nn/common/operations/internal/optimized/
Doptimized_ops.h333 void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected()
369 void FullyConnectedAsGEMV(const uint8* input_data, const Dims<4>& input_dims, in FullyConnectedAsGEMV()