Home
last modified time | relevance | path

Searched refs:bias_shape (Results 1 – 25 of 75) 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() argument
80 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4Impl() argument
120 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4Task()
129 bias_shape(bias_shape), in FullyConnectedSparseWeight1x4Task()
140 bias_shape, bias_data, output_shape, output_data, thread_start, in Run()
151 const RuntimeShape& bias_shape; member
168 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4() argument
181 bias_shape, bias_data, output_shape, output_data, 0, batches, in FullyConnectedSparseWeight1x4()
195 weights_data, bias_shape, bias_data, output_shape, 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
228 filter_shape, filter_values, filter_quantized, bias_shape, bias_values, in TF_LITE_MICRO_TEST()
239 const int bias_shape[] = {4, 1, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
271 filter_shape, filter_values, filter_quantized, bias_shape, bias_values, in TF_LITE_MICRO_TEST()
291 const int bias_shape[] = {4, 1, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
318 filter_shape, filter_values, filter_quantized, bias_shape, bias_values, in TF_LITE_MICRO_TEST()
329 const int bias_shape[] = {4, 1, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
368 filter_shape, filter_values, filter_quantized, bias_shape, bias_values, in TF_LITE_MICRO_TEST()
388 const int bias_shape[] = {4, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
410 filter_shape, filter_values, filter_quantized, bias_shape, bias_values, in TF_LITE_MICRO_TEST()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h44 const int8* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvHybridGeneral() argument
263 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridWithRounding() argument
279 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConvHybridWithRounding()
302 filter_shape, filter_data, bias_shape, bias_data, output_shape, in DepthwiseConvHybridWithRounding()
313 filter_shape, filter_data, bias_shape, bias_data, output_shape, in DepthwiseConvHybridWithRounding()
322 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridImpl() argument
329 filter_shape, filter_data, bias_shape, bias_data, output_shape, in DepthwiseConvHybridImpl()
342 const RuntimeShape& bias_shape, in DepthwiseConvHybridWorkerTask()
356 bias_shape(bias_shape), in DepthwiseConvHybridWorkerTask()
369 bias_shape, bias_data, output_shape, in Run()
[all …]
Dfully_connected.h33 const int8* filter_data, const RuntimeShape& bias_shape, in FullyConnected() argument
61 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_rows); in FullyConnected()
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc127 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvGeneral() argument
134 params, input_shape, input_data, filter_shape, filter_data, bias_shape, in DispatchDepthwiseConvGeneral()
143 const int8* filter_data, const RuntimeShape& bias_shape, in DispatchDepthwiseConvGeneral() argument
150 input_data, filter_shape, filter_data, bias_shape, bias_data, in DispatchDepthwiseConvGeneral()
163 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvImpl() argument
201 bias_shape, bias_data, output_shape, output_data, in DispatchDepthwiseConvImpl()
209 bias_shape, bias_data, output_shape, output_data, in DispatchDepthwiseConvImpl()
236 bias_shape, bias_data, output_shape, output_data, /*thread_start=*/0, in DispatchDepthwiseConvImpl()
272 bias_shape, bias_data, output_shape, output_data, /*thread_start=*/0, in DispatchDepthwiseConvImpl()
291 bias_shape, bias_data, output_shape, output_data, /*thread_start=*/0, in DispatchDepthwiseConvImpl()
[all …]
Ddepthwiseconv_float_test.cc36 const float* filter_data, const RuntimeShape& bias_shape, in TestOneDepthwiseConv() argument
42 filter_data, bias_shape, bias_data, output_shape, in TestOneDepthwiseConv()
45 params, input_shape, input_data, filter_shape, filter_data, bias_shape, in TestOneDepthwiseConv()
/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
264 input_shape, input_values, filter_shape, filter_values, bias_shape, in TF_LITE_MICRO_TEST()
277 const int bias_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
306 filter_zero_point, bias_shape, bias_values, bias_quantized, golden, in TF_LITE_MICRO_TEST()
323 const int bias_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
353 filter_zero_point, bias_shape, bias_values, bias_quantized, golden, in TF_LITE_MICRO_TEST()
364 const int bias_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
377 input_shape, input_values, filter_shape, filter_values, bias_shape, in TF_LITE_MICRO_TEST()
390 const int bias_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
417 filter_zero_point, bias_shape, bias_values, bias_quantized, golden_relu, in TF_LITE_MICRO_TEST()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc43 const TensorShape bias_shape = ctx->InputShape(1); in Compile() local
48 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(bias_shape), in Compile()
50 bias_shape.DebugString())); in Compile()
61 ctx, bias_shape.dim_size(0) == input_shape.dim_size(feature_dim), in Compile()
65 bias_shape.DebugString(), " vs. ", input_shape.DebugString())); in Compile()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h38 const T* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvWorkerTask()
47 bias_shape_(bias_shape), in DepthwiseConvWorkerTask()
122 const T* filter_data, const RuntimeShape& bias_shape, in DepthwiseConv() argument
151 filter_data, bias_shape, bias_data, output_shape, in DepthwiseConv()
175 filter_data, bias_shape, bias_data, output_shape, in DepthwiseConv()
/external/tensorflow/tensorflow/python/keras/layers/
Deinsum_dense.py146 kernel_shape, bias_shape, self.full_output_shape = shape_data
156 if bias_shape is not None:
159 shape=bias_shape,
327 bias_shape = [
333 bias_shape.append(1)
335 bias_shape = None
337 return weight_shape, bias_shape, output_shape
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_preceding_affine.cc57 const Shape& bias_shape = bias.shape(); in FuseAddOrSubParamsIntoPrecedingAffine() local
65 const int depth = bias_shape.dims(bias_shape.dimensions_count() - 1); in FuseAddOrSubParamsIntoPrecedingAffine()
69 bias_shape.dims(bias_shape.dimensions_count() - 1)) { in FuseAddOrSubParamsIntoPrecedingAffine()
127 const Shape& bias_shape = bias.shape(); in FuseMulOrDivParamsIntoPrecedingAffine() local
141 bias_shape.dims(bias_shape.dimensions_count() - 1)) { in FuseMulOrDivParamsIntoPrecedingAffine()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dfuse_bias_tf.cc87 auto bias_shape = bias.getType().dyn_cast<RankedTensorType>().getShape(); in matchAndRewrite() local
91 if (filter_shape.back() != bias_shape.back()) return failure(); in matchAndRewrite()
94 if (bias_shape.size() != 1) return failure(); in matchAndRewrite()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepthwiseconv_uint8.h95 const uint8_t* filter_data, const RuntimeShape& bias_shape, in Run()
127 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in Run()
179 const int8_t* filter_data, const RuntimeShape& bias_shape, in RunPerChannel()
214 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in RunPerChannel()
283 const uint8_t* filter_data, const RuntimeShape& bias_shape, in DepthwiseConv() argument
289 filter_data, bias_shape, in DepthwiseConv()
Dconv.h27 const float* filter_data, const RuntimeShape& bias_shape, in Conv() argument
49 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in Conv()
103 const uint8_t* filter_data, const RuntimeShape& bias_shape, in Conv() argument
132 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in Conv()
191 const RuntimeShape& bias_shape, const float* bias_data, in HybridConvPerChannel() argument
212 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in HybridConvPerChannel()
Dtranspose_conv.h28 const float* filter_data, const RuntimeShape& bias_shape, in TransposeConv() argument
51 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in TransposeConv()
117 const uint8_t* filter_data, const RuntimeShape& bias_shape, in TransposeConv() argument
149 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in TransposeConv()
Ddepthwiseconv_float.h28 const float* filter_data, const RuntimeShape& bias_shape, in DepthwiseConv() argument
54 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConv()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h26 const int8_t* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvPerChannel() argument
59 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConvPerChannel()
126 const int8_t* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvPerChannel() argument
156 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConvPerChannel()
212 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel() argument
238 const int bias_depth = bias_shape.FlatSize(); in DepthwiseConvHybridPerChannel()
Dconv.h28 const int8_t* filter_data, const RuntimeShape& bias_shape, in ConvPerChannel() argument
54 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in ConvPerChannel()
133 const int8_t* filter_data, const RuntimeShape& bias_shape, in ConvPerChannel() argument
157 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in ConvPerChannel()
Dtranspose_conv.h28 const int8_t* filter_data, const RuntimeShape& bias_shape, in TransposeConv() argument
46 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in TransposeConv()
127 const int8_t* filter_data, const RuntimeShape& bias_shape, in TransposeConv() argument
145 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in TransposeConv()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/
Dfully_connected.h29 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight() argument
40 dense_weights_data.data(), bias_shape, bias_data, output_shape, in FullyConnectedSparseWeight()
/external/tensorflow/tensorflow/python/keras/saving/
Dhdf5_format.py491 bias_shape = weights[2].shape
494 if bias_shape == (2 * units * n_gates,):
496 elif bias_shape == (units * n_gates,):
499 raise ValueError('Invalid bias shape: ' + str(bias_shape))
535 bias_shape = weights[2].shape
555 if bias_shape == (2 * units * n_gates,):
557 elif bias_shape == (2, units * n_gates):
559 elif bias_shape == (units * n_gates,):
562 raise ValueError('Invalid bias shape: ' + str(bias_shape))
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmkl_remapper_test.cc61 auto bias_shape = ops::Placeholder::Shape({128}); in FuseConv2DWithBiasAndAddNOrAdd() local
67 auto bias = Placeholder(s.WithOpName("bias"), DT_FLOAT, bias_shape); in FuseConv2DWithBiasAndAddNOrAdd()
111 TensorShape(bias_shape.shape_.dim_sizes())); in FuseConv2DWithBiasAndAddNOrAdd()
227 auto bias_shape = Placeholder::Shape({3}); \
232 auto bias = Placeholder(s.WithOpName("bias"), DT_FLOAT, bias_shape); \
458 auto bias_shape = ops::Placeholder::Shape({8}); in TEST_F() local
464 auto bias = Placeholder(s.WithOpName("bias"), DT_FLOAT, bias_shape); in TEST_F()
481 TensorShape(bias_shape.shape_.dim_sizes())); in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dfully_connected_tester.cc211 const std::array<int32_t, 1> bias_shape{{OutputChannels()}}; in CreateTfLiteModel() local
222 builder.CreateVector<int32_t>(bias_shape.data(), bias_shape.size()), in CreateTfLiteModel()
235 builder.CreateVector<int32_t>(bias_shape.data(), bias_shape.size()), in CreateTfLiteModel()
Dconv_2d_tester.cc247 const std::array<int32_t, 1> bias_shape{{OutputChannels()}}; in CreateTfLiteModel() local
282 builder.CreateVector<int32_t>(bias_shape.data(), bias_shape.size()), in CreateTfLiteModel()
295 builder.CreateVector<int32_t>(bias_shape.data(), bias_shape.size()), in CreateTfLiteModel()
Ddepthwise_conv_2d_tester.cc240 const std::array<int32_t, 1> bias_shape{{OutputChannels()}}; in CreateTfLiteModel() local
250 builder.CreateVector<int32_t>(bias_shape.data(), bias_shape.size()), in CreateTfLiteModel()
286 builder.CreateVector<int32_t>(bias_shape.data(), bias_shape.size()), in CreateTfLiteModel()

123