| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op_gpu.cu.cc | 57 T* output, int32 bias_size) { in BiasNHWCKernel() argument
59 int32 bias_offset = index % bias_size; in BiasNHWCKernel()
66 T* output, int32 bias_size, int32 image_size) { in BiasNCHWKernel() argument
69 int32 bias_offset = index2 % bias_size; in BiasNCHWKernel()
80 const int32 bias_size = channel; in compute() local
82 const int32 total_count = batch * bias_size * image_size; in compute()
91 output, bias_size)); in compute()
96 output, bias_size, image_size)); in compute()
103 T* bias_backprop, int32 bias_size) { in BiasGradNHWC_Naive() argument
105 int32 bias_offset = index % bias_size; in BiasGradNHWC_Naive()
[all …]
|
| D | bias_op.h | 34 const int64_t bias_size = bias.dimension(0); in operator() local
35 const int64_t rest_size = input.size() / bias_size; in operator()
41 const int bias_size = bias.dimension(0); in operator() local
42 const int rest_size = input.size() / bias_size; in operator()
|
| D | mkl_fused_ops_test.cc | 105 const int bias_size = filter_count; in VerifyBiasAddTensorsClose() local
106 Tensor bias(dtype, {bias_size}); in VerifyBiasAddTensorsClose()
|
| D | conv_ops_test.cc | 759 const int bias_size = filter_count; in VerifyBiasAddTensorsNear() local
760 Tensor bias(dtype, {bias_size}); in VerifyBiasAddTensorsNear()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | depthwise_conv_test.cc | 52 int bias_size = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel() local
54 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseDepthwiseConvolutionOpModel()
72 {bias_size}, in BaseDepthwiseConvolutionOpModel()
85 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseDepthwiseConvolutionOpModel()
|
| D | conv_test.cc | 54 int bias_size = GetShape(filter_)[0]; in BaseConvolutionOpModel() local
56 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseConvolutionOpModel()
74 {bias_size}, in BaseConvolutionOpModel()
87 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseConvolutionOpModel()
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | neon_depthwise_conv_op.cc | 130 int64 bias_size = bias_neon_dims.sizes[0]; in Compute() local
132 bias_size * sizeof(float), Allocator::kAllocatorAlignment)); in Compute()
133 memset(bias_ptr, 0, bias_size * sizeof(float)); in Compute()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op_test_base.py | 231 bias_size = filter_in_sizes[-1] # equals to output depth
245 t3 = constant_op.constant(x3, shape=[bias_size], dtype=dtype)
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 275 int bias_size = GetShape(filter_)[0]; in ConvolutionOpModel() local
277 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in ConvolutionOpModel()
283 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in ConvolutionOpModel()
454 int bias_size = GetShape(filter_)[3]; in DepthwiseConvolutionOpModel() local
456 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in DepthwiseConvolutionOpModel()
462 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in DepthwiseConvolutionOpModel()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 195 const int bias_size = bias_shape.FlatSize(); in AddBiasAndEvalActivationFunction() local
197 TFLITE_DCHECK_EQ((array_size % bias_size), 0); in AddBiasAndEvalActivationFunction()
202 for (; array_ptr != array_end_ptr; array_ptr += bias_size) { in AddBiasAndEvalActivationFunction()
204 for (; i <= bias_size - 16; i += 16) { in AddBiasAndEvalActivationFunction()
230 for (; i <= bias_size - 4; i += 4) { in AddBiasAndEvalActivationFunction()
238 for (; i < bias_size; i++) { in AddBiasAndEvalActivationFunction()
246 const int bias_size = bias_shape.FlatSize(); in AddBiasAndEvalActivationFunction()
248 TFLITE_DCHECK_EQ((array_size % bias_size), 0); in AddBiasAndEvalActivationFunction()
250 array_offset += bias_size) { in AddBiasAndEvalActivationFunction()
251 for (int i = 0; i < bias_size; i++) { in AddBiasAndEvalActivationFunction()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 290 const int bias_size = bias_shape.FlatSize(); in AddBiasAndEvalActivationFunction() local
292 TFLITE_DCHECK_EQ((array_size % bias_size), 0); in AddBiasAndEvalActivationFunction()
297 for (; array_ptr != array_end_ptr; array_ptr += bias_size) { in AddBiasAndEvalActivationFunction()
299 for (; i <= bias_size - 16; i += 16) { in AddBiasAndEvalActivationFunction()
325 for (; i <= bias_size - 4; i += 4) { in AddBiasAndEvalActivationFunction()
333 for (; i < bias_size; i++) { in AddBiasAndEvalActivationFunction()
341 const int bias_size = bias_shape.FlatSize(); in AddBiasAndEvalActivationFunction()
343 TFLITE_DCHECK_EQ((array_size % bias_size), 0); in AddBiasAndEvalActivationFunction()
345 array_offset += bias_size) { in AddBiasAndEvalActivationFunction()
346 for (int i = 0; i < bias_size; i++) { in AddBiasAndEvalActivationFunction()
|