Home
last modified time | relevance | path

Searched refs:bias_shape_ (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h47 bias_shape_(bias_shape), in DepthwiseConvWorkerTask()
58 filter_data_, bias_shape_, bias_data_, output_shape_, in Run()
69 const RuntimeShape& bias_shape_; member
Dlegacy_optimized_ops.h393 bias_shape_(bias_shape), in LegacyDepthwiseConvWorkerTask()
403 filter_data_, bias_shape_, bias_data_, in Run()
414 const RuntimeShape& bias_shape_; member
505 bias_shape_(bias_shape), in LegacyPerChannelDepthwiseConvWorkerTask()
517 filter_shape_, filter_data_, bias_shape_, bias_data_, output_shape_, in Run()
529 const RuntimeShape& bias_shape_; member
1004 bias_shape_(bias_shape), in LegacyFullyConnectedAsGEMVWorkerTask()
1019 filter_offset_, bias_shape_, bias_data_, output_offset_, in Run()
1031 const RuntimeShape& bias_shape_; member
2015 bias_shape_(bias_shape), in LegacyInt8FullyConnectedAsGEMVWorkerTask()
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dmatmul_builder.h41 std::vector<int> weights_shape_, bias_shape_; variable
68 std::vector<int> weights_shape_, bias_shape_; variable
Dtranspose_conv_2d_builder.h44 std::vector<int> bias_shape_; variable
Dtranspose_conv_2d_builder.cc131 bias_shape_ = {1, 1, 1, output_depth_size}; in PopulateSubGraph()
133 bias_shape_.data(), reinterpret_cast<char*>(bias_data.data()), in PopulateSubGraph()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv.h1911 bias_shape_(bias_shape),
1922 input_data_, filter_shape_, filter_data_, bias_shape_,
1935 const RuntimeShape& bias_shape_;