Searched refs:bias_shape_ (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | depthwiseconv_multithread.h | 47 bias_shape_(bias_shape), in DepthwiseConvWorkerTask() 58 filter_data_, bias_shape_, bias_data_, output_shape_, in Run() 69 const RuntimeShape& bias_shape_; member
|
D | legacy_optimized_ops.h | 393 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/ |
D | matmul_builder.h | 41 std::vector<int> weights_shape_, bias_shape_; variable 68 std::vector<int> weights_shape_, bias_shape_; variable
|
D | transpose_conv_2d_builder.h | 44 std::vector<int> bias_shape_; variable
|
D | transpose_conv_2d_builder.cc | 131 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/ |
D | depthwise_conv.h | 1911 bias_shape_(bias_shape), 1922 input_data_, filter_shape_, filter_data_, bias_shape_, 1935 const RuntimeShape& bias_shape_;
|