Home
last modified time | relevance | path

Searched refs:bias_size (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbias_op_gpu.cu.cc49 T* output, int32 bias_size) { in BiasNHWCKernel() argument
51 int32 bias_offset = index % bias_size; in BiasNHWCKernel()
58 T* output, int32 bias_size, int32 image_size) { in BiasNCHWKernel() argument
61 int32 bias_offset = index2 % bias_size; in BiasNCHWKernel()
72 const int32 bias_size = channel; in compute() local
74 const int32 total_count = batch * bias_size * image_size; in compute()
82 config.virtual_thread_count, input, bias, output, bias_size); in compute()
86 config.virtual_thread_count, input, bias, output, bias_size, in compute()
94 T* bias_backprop, int32 bias_size) { in BiasGradNHWC_Naive() argument
96 int32 bias_offset = index % bias_size; in BiasGradNHWC_Naive()
[all …]
Dbias_op.h34 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()
Dmkl_conv_ops.cc600 memory::dims bias_size; in Compute() local
601 conv_utl.GetBiasSizeInMklOrder(kInputIndex_Bias, &bias_size); in Compute()
603 bias.SetUsrMem(bias_size, memory::format::x, &bias_tensor); in Compute()
604 bias.SetOpMemDesc(bias_size, memory::format::any); in Compute()
/external/tensorflow/tensorflow/contrib/bayesflow/python/kernel_tests/
Dlayers_conv_variational_test.py199 bias_size = (filters,)
201 result_log_prob=random_ops.random_uniform(bias_size, seed=seed()),
202 result_sample=random_ops.random_uniform(bias_size, seed=seed()))
204 result_log_prob=random_ops.random_uniform(bias_size, seed=seed()),
205 result_sample=random_ops.random_uniform(bias_size, seed=seed()))
207 result=random_ops.random_uniform(bias_size, seed=seed()))
414 bias_size = (filters,)
427 bias_size, seed=seed()),
429 bias_size, seed=seed()),
431 bias_size, seed=seed()),
[all …]
Dlayers_dense_variational_test.py164 bias_size = [out_size]
166 result_log_prob=random_ops.random_uniform(bias_size, seed=seed()),
167 result_sample=random_ops.random_uniform(bias_size, seed=seed()))
169 result_log_prob=random_ops.random_uniform(bias_size, seed=seed()),
170 result_sample=random_ops.random_uniform(bias_size, seed=seed()))
172 result=random_ops.random_uniform(bias_size, seed=seed()))
/external/tensorflow/tensorflow/contrib/lite/kernels/
Ddepthwise_conv_test.cc37 int bias_size = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel() local
39 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseDepthwiseConvolutionOpModel()
45 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseDepthwiseConvolutionOpModel()
Dconv_test.cc53 int bias_size = GetShape(filter_)[0]; in BaseConvolutionOpModel() local
55 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseConvolutionOpModel()
61 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseConvolutionOpModel()
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc130 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/
Dfused_conv2d_bias_activation_op_test.py188 bias_size = filter_in_sizes[-1] # equals to output depth
202 t3 = constant_op.constant(x3, shape=[bias_size], dtype=dtype)
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h237 const int bias_size = bias_dims.sizes[3] * bias_dims.strides[3]; in AddBiasAndEvalActivationFunction() local
239 TFLITE_DCHECK_EQ((array_size % bias_size), 0); in AddBiasAndEvalActivationFunction()
244 for (; array_ptr != array_end_ptr; array_ptr += bias_size) { in AddBiasAndEvalActivationFunction()
246 for (; i <= bias_size - 16; i += 16) { in AddBiasAndEvalActivationFunction()
272 for (; i <= bias_size - 4; i += 4) { in AddBiasAndEvalActivationFunction()
280 for (; i < bias_size; i++) { in AddBiasAndEvalActivationFunction()
288 const int bias_size = bias_dims.sizes[3] * bias_dims.strides[3]; in AddBiasAndEvalActivationFunction()
290 TFLITE_DCHECK_EQ((array_size % bias_size), 0); in AddBiasAndEvalActivationFunction()
292 array_offset += bias_size) { in AddBiasAndEvalActivationFunction()
293 for (int i = 0; i < bias_size; i++) { in AddBiasAndEvalActivationFunction()