| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_fused_conv_rewriter.cc | 41 HloInstruction* side_input; member
77 HloInstruction* side_input = nullptr; in FindConvWithRelu() local
|
| D | gpu_conv_runner.cc | 109 se::DeviceMemory<OutputType> side_input(params.fusion->side_input_buf); in RunGpuConvForwardActivation() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 159 Output side_input = ops::Const(root.WithOpName("side_input"), in RunFusedBatchNorm() local
251 Output side_input = ops::Const(root.WithOpName("side_input"), in RunFusedBatchNormEx() local
364 Tensor side_input(t_dtype, input_shape); in VerifyTensorsNear() local
|
| D | fused_batch_norm_op.cu.cc | 334 const T* side_input, float epsilon, T* out) { in FusedBatchNormInferenceMetaKernel()
|
| D | fused_batch_norm_op.cc | 1161 const Tensor* side_input = has_side_input_ ? &context->input(5) : nullptr; in ComputeWithReservedSpace() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 94 int side_input = kMissingIndex; member
1242 const NodeDef& side_input = graph->node(matched.side_input); in AddFusedBatchNormExNode() local
|
| D | remapper_test.cc | 253 auto side_input = Placeholder(s.WithOpName("side_input"), DT_FLOAT, in TEST_F() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 258 auto side_input = op_context.op_info.add_inputs(); in DescribeFusedConv2DBiasActivation() local
|
| D | op_level_cost_estimator.cc | 1316 auto& side_input = op_context.op_info.inputs(3); in PredictFusedConv2DBiasActivation() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3276 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3299 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3324 const DeviceMemory<U>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3464 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3490 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3518 const DeviceMemory<U>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 345 const DeviceMemory<float> &side_input, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
404 const DeviceMemory<float> &side_input, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3164 TRT_ShapedWeights side_input = inputs.at(3).weights(); in ConvertFusedConv2DBiasActivation() local
|