Home
last modified time | relevance | path

Searched defs:bias_shape (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h32 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
68 const uint8* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
117 const uint8* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
171 const uint8* shuffled_weights_data, const RuntimeShape& bias_shape, in ShuffledFullyConnected()
Ddepthwiseconv_float.h28 const float* filter_data, const RuntimeShape& bias_shape, in DepthwiseConv()
Ddepthwiseconv_uint8.h181 const uint8* filter_data, const RuntimeShape& bias_shape, in DepthwiseConv()
Dreference_ops.h135 const float* filter_data, const RuntimeShape& bias_shape, in Conv()
208 const uint8* filter_data, const RuntimeShape& bias_shape, in Conv()
1858 const RuntimeShape bias_shape = in LstmCell() local
2070 const RuntimeShape bias_shape = in LstmCell() local
/external/tensorflow/tensorflow/contrib/fused_conv/ops/
Dfused_conv2d_bias_activation_op.cc74 ShapeHandle bias_shape; in __anon0cecbc340102() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_preceding_affine.cc44 const Shape& bias_shape = bias.shape(); in FuseAddOrSubParamsIntoPrecedingAffine() local
113 const Shape& bias_shape = bias.shape(); in FuseMulOrDivParamsIntoPrecedingAffine() local
Dfuse_binary_into_following_affine.cc71 const Shape& bias_shape = bias.shape(); in FuseAddOrSubParamsIntoFollowingAffine() local
Dpropagate_fixed_sizes.cc1026 const auto& bias_shape = bias_array.shape(); in ProcessLstmCellOperator() local
1155 const auto& bias_shape = bias_array.shape(); in ProcessUnidirectionalSequenceRnnOperator() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc43 const TensorShape bias_shape = ctx->InputShape(1); in Compile() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h26 const int8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
Dconv.h28 const int8* filter_data, const RuntimeShape& bias_shape, in ConvPerChannel()
Ddepthwise_conv.h26 const int8* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc109 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConv()
293 const RuntimeShape& bias_shape, int stride, PaddingType padding_type, in TestOneDepthwiseConvWithGivenOutputShift()
426 const RuntimeShape& bias_shape, int stride, PaddingType padding_type, in TestOneDepthwiseConvBisectOutputShift()
476 const RuntimeShape& bias_shape, int stride, PaddingType padding_type, in TestOneDepthwiseConv()
Ddepthwiseconv_float_test.cc35 const float* filter_data, const RuntimeShape& bias_shape, in TestOneDepthwiseConv()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc113 auto bias_shape = ops::Placeholder::Shape({128}); in TEST_F() local
176 auto bias_shape = Placeholder::Shape({128}); in TEST_F() local
397 auto bias_shape = ops::Placeholder::Shape({128}); in TEST_F() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dmultithreaded_conv.h138 const float* filter_data, const RuntimeShape& bias_shape, in Conv()
Doptimized_ops.h189 const RuntimeShape& bias_shape, in AddBiasAndEvalActivationFunction()
297 const RuntimeShape& bias_shape, in GEMVForLstmCell()
485 const RuntimeShape& bias_shape, const int32* bias_data, in GEMVForLstmCellWithSymmetricRange()
776 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
820 const RuntimeShape& bias_shape, const int32* bias_data, int32 output_offset, in FullyConnectedAsGEMVWorkerImpl()
1096 const RuntimeShape& bias_shape, const int32* bias_data, int32 output_offset, in FullyConnectedAsGEMV()
1169 const uint8* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
1228 const uint8* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
1634 const uint8* shuffled_weights_data, const RuntimeShape& bias_shape, in ShuffledFullyConnected()
2182 const float* filter_data, const RuntimeShape& bias_shape, in Conv()
[all …]
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad_test.cc220 TensorShape bias_shape({5}); in TEST_F() local
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc99 const RuntimeShape& bias_shape, const int32* bias_data, in DepthwiseConvOptimizedForFilterWidthEight()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h284 const RuntimeShape& bias_shape, in AddBiasAndEvalActivationFunction()
419 const RuntimeShape& bias_shape, in GEMVForLstmCell()
607 const RuntimeShape& bias_shape, const int32* bias_data, in GEMVForLstmCellWithSymmetricRange()
898 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
965 const RuntimeShape& bias_shape, const int32* bias_data, int32 output_offset, in FullyConnectedAsGEMV()
1152 const uint8* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
1236 const uint8* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
1690 const uint8* shuffled_weights_data, const RuntimeShape& bias_shape, in ShuffledFullyConnected()
2113 const float* filter_data, const RuntimeShape& bias_shape, in Conv()
2206 const RuntimeShape& bias_shape, const float* bias_data, in HybridConv()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc249 ShapeHandle bias_shape; in BiasAddShape() local