| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| fully_connected_test.cc | 32 std::initializer_list<float> bias_data,
49 CreateFloatTensor(bias_data, bias_dims, "bias_tensor"),
107 std::initializer_list<int32_t> bias_data, float bias_min, float bias_max,
125 CreateQuantized32Tensor(bias_data, bias_dims, "bias_tensor", bias_min,
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| depthwiseconv_float.h | [all...] |
| /external/tensorflow/tensorflow/lite/toco/ |
| export_tensorflow.cc | 442 const float* bias_data = local
444 ConvertFloatTensorConst(bias_array_name, bias_shape_1d, bias_data,
532 const float* bias_data = local
534 ConvertFloatTensorConst(bias_name, bias_shape_1d, bias_data,
666 const float* bias_data = local
669 bias_shape_1d, bias_data, AxesOrder::kOneAxis, local
1515 const float* bias_data = local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_conv_ops.cc | 118 // bias_data: input data buffer of bias
121 const Tbias* bias_data, const Toutput* dst_data) {
127 static_cast<void*>(const_cast<Tbias*>(bias_data)));
1148 Tbias* bias_data = variable
[all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_dnn.h | 540 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode,
[all...] |
| rocm_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| reference_ops.h | 136 const float* bias_data, const RuntimeShape& output_shape,
156 if (bias_data) {
193 if (bias_data) {
194 bias_value = bias_data[out_channel];
209 const int32* bias_data, const RuntimeShape& output_shape,
236 if (bias_data) {
273 if (bias_data) {
274 acc += bias_data[out_channel];
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | [all...] |
