| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_fused_conv_rewriter.cc | 37 HloInstruction* side_input; member
72 HloInstruction* side_input = nullptr; in FindConvWithRelu() local
95 auto side_input_pattern = Op(&side_input); in FindConvWithRelu()
124 } else if (side_input == nullptr && Match(addend, side_input_pattern)) { in FindConvWithRelu()
125 CHECK(side_input); in FindConvWithRelu()
167 side_input, in FindConvWithRelu()
193 if (match.side_input) { in TryRewriteToCudnnForwardRelu()
221 if (match.side_input) { in TryRewriteToCudnnForwardRelu()
222 args.push_back(match.side_input); in TryRewriteToCudnnForwardRelu()
|
| D | cudnn_conv_runner.cc | 281 se::DeviceMemory<T> side_input(params.fusion->side_input_buf); in RunCudnnConvImpl() local
283 if (side_input.is_null()) { in RunCudnnConvImpl()
295 side_input = output_buf; in RunCudnnConvImpl()
300 filter_descriptor, filter_buf, convolution_descriptor, side_input, in RunCudnnConvImpl()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op.py | 37 side_input=None, argument
96 if side_input is None:
97 side_input = []
102 side_input,
|
| D | fused_conv2d_bias_activation_op_test_base.py | 672 side_input, biases, apply_relu): argument
700 gen_array_ops.dequantize(side_input, -128, 127))
897 side_input, _, _ = gen_array_ops.quantize_v2(
921 side_input=side_input,
928 side_input_scale, side_input, biases, apply_relu)
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 196 const Tensor& side_input = context->input(kSideInput); in Compute() local
198 OP_REQUIRES_OK(context, CheckShape(side_input, "side_input")); in Compute()
242 side_input, side_input_scale, bias, activation_mode_, in Compute()
314 const Tensor& bias, const Tensor* side_input, in LogFusedConvAutotuneResults() argument
328 if (side_input) { in LogFusedConvAutotuneResults()
329 side_input->shape().AsProto( in LogFusedConvAutotuneResults()
331 instr.mutable_side_input()->set_dtype(side_input->dtype()); in LogFusedConvAutotuneResults()
544 const Tensor* side_input = &side_input_param; in launch() local
559 output_depth, &maybe_transformed_side_input, &side_input))); in launch()
648 side_input->template flat<T>().data()), in launch()
[all …]
|
| D | fused_conv2d_bias_activation_op.h | 42 const Eigen::PaddingType& padding, const Tensor& side_input,
56 const Eigen::PaddingType& padding, const Tensor& side_input,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 82 const Tensor& bias, const Tensor* side_input, in LogFusedConvAutotuneResults() argument
96 if (side_input) { in LogFusedConvAutotuneResults()
97 side_input->shape().AsProto( in LogFusedConvAutotuneResults()
99 instr.mutable_side_input()->set_dtype(side_input->dtype()); in LogFusedConvAutotuneResults()
|
| D | gpu_utils.h | 174 const Tensor& bias, const Tensor* side_input,
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | conv_autotuning.proto | 17 TensorProto side_input = 6; field
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | rev_block_lib_test.py | 168 def f(x, side_input): argument
170 x, self.CHANNELS // 2, use_bias=True) + side_input[0]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1177 auto& side_input = op_context.op_info.inputs(3); in PredictFusedConv2DBiasActivation() local
1205 if (side_input.shape().dim_size() > 0) { in PredictFusedConv2DBiasActivation()
1206 component_ops.push_back(FusedChildContext(op_context, "Mul", side_input, in PredictFusedConv2DBiasActivation()
1207 {side_input, side_input_scale})); in PredictFusedConv2DBiasActivation()
1209 FusedChildContext(op_context, "Add", output, {side_input, output})); in PredictFusedConv2DBiasActivation()
|
| D | op_level_cost_estimator_test.cc | 225 auto side_input = op_context.op_info.add_inputs(); in DescribeFusedConv2DBiasActivation() local
228 DescribeTensor4D(batch, ox, oy, oz, side_input); in DescribeFusedConv2DBiasActivation()
230 DescribeTensor4D(batch, oz, ox, oy, side_input); in DescribeFusedConv2DBiasActivation()
|