| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | gpu_fusion.cc | 71 Value side_input; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cu.cc | 205 const IT* side_input = reinterpret_cast<const IT*>(_side_input); in run() local
299 const T* side_input, float epsilon, T* out) { in FusedBatchNormInferenceMetaKernel()
|
| 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.cc | 1270 const Tensor* side_input = has_side_input_ ? &context->input(5) : nullptr; in ComputeWithReservedSpace() local
|
| /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 | 123 se::DeviceMemory<OutputType> side_input(params.fusion->side_input_buf); in RunGpuConvForwardActivation() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | mkl_remapper_test.cc | 342 auto side_input = in TEST_F() local
|
| D | remapper.cc | 107 int side_input = kMissingIndex; member
1471 const NodeDef& side_input = graph->node(matched.side_input); in AddFusedBatchNormExNode() local
|
| D | remapper_test.cc | 255 auto side_input = Placeholder(s.WithOpName("side_input"), DT_FLOAT, in TEST_F() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1124 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
1143 const DeviceMemory<Eigen::half>& side_input, in DoBatchNormalizationForward()
|
| D | stream.cc | 347 const DeviceMemory<float> &side_input, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
397 const DeviceMemory<Eigen::half> &side_input, in ThenBatchNormalizationForward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3520 const DeviceMemory<Eigen::half>& side_input, in DoBatchNormalizationForward()
3541 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3563 const DeviceMemory<T>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 4562 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
4585 const DeviceMemory<Eigen::half>& side_input, in DoBatchNormalizationForward()
4611 const DeviceMemory<T>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 280 auto side_input = op_context.op_info.add_inputs(); in DescribeFusedConv2DBiasActivation() local
|
| D | op_level_cost_estimator.cc | 1757 auto& side_input = op_context.op_info.inputs(3); in PredictFusedConv2DBiasActivation() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3393 TRT_ShapedWeights side_input = inputs.at(3).weights(); in ConvertFusedConv2DBiasActivation() local
|