Home
last modified time | relevance | path

Searched refs:bias_tensor (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/
Dsubgraph_quantizer_test.cc82 const auto bias_tensor = in TEST() local
91 EXPECT_EQ(bias_tensor->type, TensorType_INT32); in TEST()
97 ASSERT_TRUE(bias_tensor->quantization); in TEST()
99 const std::vector<float>& bias_scales = bias_tensor->quantization->scale; in TEST()
122 const auto bias_buffer = model.buffers[bias_tensor->buffer].get(); in TEST()
123 ASSERT_EQ(bias_buffer->data.size(), sizeof(int32_t) * bias_tensor->shape[0]); in TEST()
127 readonly_model->buffers()->Get(bias_tensor->buffer); in TEST()
132 for (size_t i = 0; i < bias_tensor->shape[0]; i++) { in TEST()
181 const auto bias_tensor = in TEST() local
190 EXPECT_EQ(bias_tensor->type, TensorType_INT32); in TEST()
[all …]
Dsubgraph_quantizer.cc252 TensorT* bias_tensor = subgraph_->tensors[bias_tensor_idx].get(); in QuantizeOpWithBias() local
255 bias_tensor, error_reporter_)); in QuantizeOpWithBias()
/external/tensorflow/tensorflow/python/kernel_tests/
Dbias_op_test.py147 bias_tensor = constant_op.constant(bias, shape=bias.shape, dtype=dtype)
149 input_tensor, bias_tensor, data_format=data_format)
153 bias_tensor, bias.shape, output_tensor, np_input.shape)
157 nn_ops.l2_loss(output_tensor), bias_tensor)[0]
169 bias_tensor = constant_op.constant(
172 input_tensor, bias_tensor, data_format=data_format)
177 _, bias_jacob_n = gradient_checker.compute_gradient(bias_tensor,
183 nn_ops.l2_loss(output_tensor), bias_tensor)[0]
/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.cc1147 const Tensor& bias_tensor = MklGetInput(context, kInputIndex_Bias); in Compute() local
1149 this->GetBiasHandle(context, conv_fwd_pd, bias_tensor); in Compute()
1241 const Tensor& bias_tensor) { in GetBiasHandle() argument
1244 const_cast<Tbias*>(bias_tensor.flat<Tbias>().data())); in GetBiasHandle()
1626 const Tensor& bias_tensor) override { in GetBiasHandle() argument
1643 const_cast<Tbias*>(bias_tensor.flat<Tbias>().data())); in GetBiasHandle()
1656 const_cast<Tbias*>(bias_tensor.flat<Tbias>().data())); in GetBiasHandle()
/external/tensorflow/tensorflow/contrib/quantize/python/
Dfold_batch_norms.py87 bias_tensor = math_ops.subtract(
133 new_layer_tensor, bias_tensor, name='add_fold')