Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h34 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight()
80 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4Impl()
151 const RuntimeShape& bias_shape; member
168 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4()
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Ddepthwise_conv_slicing_test.cc205 const int bias_shape[] = {4, 1, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
239 const int bias_shape[] = {4, 1, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
291 const int bias_shape[] = {4, 1, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
329 const int bias_shape[] = {4, 1, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
388 const int bias_shape[] = {4, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
421 const int bias_shape[] = {4, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
477 const int bias_shape[] = {4, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
510 const int bias_shape[] = {4, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddepthwise_conv_test.cc249 const int bias_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
277 const int bias_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
323 const int bias_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
364 const int bias_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
390 const int bias_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
433 const int bias_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
474 const int bias_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
514 const int bias_shape[] = {4, 1, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
559 const int bias_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
602 const int bias_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h29 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
65 const uint8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
113 const uint8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
167 const uint8_t* shuffled_weights_data, const RuntimeShape& bias_shape, in ShuffledFullyConnected()
Dsvdf.h70 const int16_t* weights_time_data, const RuntimeShape& bias_shape, in EvalIntegerSVDF()
156 const float* weights_time_data, const RuntimeShape& bias_shape, in EvalFloatSVDF()
196 const RuntimeShape& bias_shape, const float* bias_data, float* scratch, in EvalHybridSVDF()
Dconv.h27 const float* filter_data, const RuntimeShape& bias_shape, in Conv()
103 const uint8_t* filter_data, const RuntimeShape& bias_shape, in Conv()
191 const RuntimeShape& bias_shape, const float* bias_data, in HybridConvPerChannel()
Dtranspose_conv.h28 const float* filter_data, const RuntimeShape& bias_shape, in TransposeConv()
117 const uint8_t* filter_data, const RuntimeShape& bias_shape, in TransposeConv()
Ddepthwiseconv_float.h28 const float* filter_data, const RuntimeShape& bias_shape, in DepthwiseConv()
Dconv3d.h26 const float* filter_data, const RuntimeShape& bias_shape, in Conv3D()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h26 const int8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
68 const int8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
Ddepthwise_conv.h26 const int8_t* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvPerChannel()
126 const int8_t* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvPerChannel()
212 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel()
Dconv.h28 const int8_t* filter_data, const RuntimeShape& bias_shape, in ConvPerChannel()
133 const int8_t* filter_data, const RuntimeShape& bias_shape, in ConvPerChannel()
Dtranspose_conv.h28 const int8_t* filter_data, const RuntimeShape& bias_shape, in TransposeConv()
127 const int8_t* filter_data, const RuntimeShape& bias_shape, in TransposeConv()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h44 const int8* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvHybridGeneral()
263 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridWithRounding()
322 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridImpl()
381 const RuntimeShape& bias_shape; member
396 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel()
Dfully_connected.h33 const int8* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
Dtranspose_conv.h28 const int8_t* hwoi_ordered_filter_data, const RuntimeShape& bias_shape, in TransposeConvV2()
Dconv.h35 const int8* filter_data, const RuntimeShape& bias_shape, in ConvPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/
Dfully_connected.h29 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight()
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc127 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvGeneral()
143 const int8* filter_data, const RuntimeShape& bias_shape, in DispatchDepthwiseConvGeneral()
163 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvImpl()
366 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvImpl()
533 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConv()
618 const std::int32_t* bias_data, const RuntimeShape& bias_shape, int stride, in TestOneDepthwiseConvWithGivenOutputShift()
755 const std::int32_t* bias_data, const RuntimeShape& bias_shape, int stride, in TestOneDepthwiseConvBisectOutputShift()
814 const std::int32_t* bias_data, const RuntimeShape& bias_shape, int stride, in TestOneDepthwiseConv()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_preceding_affine.cc57 const Shape& bias_shape = bias.shape(); in FuseAddOrSubParamsIntoPrecedingAffine() local
127 const Shape& bias_shape = bias.shape(); in FuseMulOrDivParamsIntoPrecedingAffine() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc43 const TensorShape bias_shape = ctx->InputShape(1); in Compile() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dfuse_bias_tf.cc87 auto bias_shape = bias.getType().dyn_cast<RankedTensorType>().getShape(); in matchAndRewrite() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc338 auto bias_shape = ops::Placeholder::Shape({128}); in TEST_F() local
403 auto bias_shape = ops::Placeholder::Shape({64}); in RunTest() local
481 auto bias_shape = Placeholder::Shape({128}); in TEST_F() local
552 auto bias_shape = Placeholder::Shape({128}); in TEST_F() local
652 auto bias_shape = ops::Placeholder::Shape({64}); in RunTest() local
944 auto bias_shape = ops::Placeholder::Shape({128}); in TEST_F() local
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dconv.cc39 const RuntimeShape& bias_shape, const int32_t* bias_data, in ConvPerChannel()
174 const RuntimeShape& bias_shape, const int32_t* bias_data, in Conv1x32Input32x32Filter()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dmultithreaded_conv.h138 const float* filter_data, const RuntimeShape& bias_shape, in Conv()

123