| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 174 const Tensor& conv_input = context->input(kConvInput); in Compute() local
175 OP_REQUIRES_OK(context, CheckShape(conv_input, "conv_input")); in Compute()
207 const int32 batch_size = GetTensorDim(conv_input, data_format_, 'N'); in Compute()
208 const int32 conv_input_rows = GetTensorDim(conv_input, data_format_, 'H'); in Compute()
209 const int32 conv_input_cols = GetTensorDim(conv_input, data_format_, 'W'); in Compute()
240 launcher_.launch(context, cudnn_use_autotune_, conv_input, conv_input_scale, in Compute()
483 const Tensor* conv_input = &conv_input_param; in launch() local
536 conv_input = &maybe_padded_conv_input; in launch()
552 ctx, *conv_input, batch_size, conv_input_rows, in launch()
554 &maybe_transformed_conv_input, &conv_input))); in launch()
[all …]
|
| D | fused_conv2d_bias_activation_op.h | 40 const Tensor& conv_input, ScaleType conv_input_scale,
54 const Tensor& conv_input, ScaleType conv_input_scale,
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op.py | 30 def fused_conv2d_bias_activation(conv_input, argument
99 conv_input,
|
| D | fused_conv2d_bias_activation_op_test_base.py | 670 def _SimulateFusedConv2dBiasActivationInt8(conv_input_scale, conv_input, kernel, argument
693 _NchwVectCToNchw(gen_array_ops.dequantize(conv_input, -128, 127)),
873 conv_input, _, _ = gen_array_ops.quantize_v2(
914 conv_input,
927 conv_input_scale, conv_input, kernel, padding_type, strides,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 246 xla::XlaOp conv_input, in MakeXlaForwardConvOp() argument
251 auto* builder = conv_input.builder(); in MakeXlaForwardConvOp()
252 TF_ASSIGN_OR_RETURN(xla::Shape input_shape, builder->GetShape(conv_input)); in MakeXlaForwardConvOp()
319 conv_input, filter, window_strides, padding, lhs_dilation, rhs_dilation, in MakeXlaForwardConvOp()
|
| D | conv_op_helpers.h | 57 xla::XlaOp conv_input,
|
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | quantize_test.py | 294 conv_input = array_ops.zeros((batch_size, height, width, depth))
296 conv_input,
307 conv_input,
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1174 auto& conv_input = op_context.op_info.inputs(0); in PredictFusedConv2DBiasActivation() local
1184 conv_input.shape(), filter.shape(), op_context.op_info, in PredictFusedConv2DBiasActivation()
1199 FusedChildContext(op_context, "Conv2D", output, {conv_input, filter}), in PredictFusedConv2DBiasActivation()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 1714 conv_input = input_tensor + 0.1
1719 conv_input,
1726 conv_input,
|