/external/tensorflow/tensorflow/core/kernels/ |
D | bias_op_gpu.cu.cc | 49 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 …]
|
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_conv_ops.cc | 600 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/ |
D | layers_conv_variational_test.py | 199 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 …]
|
D | layers_dense_variational_test.py | 164 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/ |
D | depthwise_conv_test.cc | 37 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()
|
D | conv_test.cc | 53 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/ |
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.py | 188 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/ |
D | optimized_ops.h | 237 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()
|