| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | gpu_fusion.cc | 70 Value side_input; in matchAndRewrite() 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 | 111 se::DeviceMemory<OutputType> side_input(params.fusion->side_input_buf); in RunGpuConvForwardActivation() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cu.cc | 207 const IT* side_input = reinterpret_cast<const IT*>(_side_input); in run() local
301 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 | 1249 const Tensor* side_input = has_side_input_ ? &context->input(5) : nullptr; in ComputeWithReservedSpace() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | mkl_remapper_test.cc | 343 auto side_input = in TEST_F() local
|
| D | remapper.cc | 105 int side_input = kMissingIndex; member
1413 const NodeDef& side_input = graph->node(matched.side_input); in IsCpuCompatibleDataType() 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 | 1016 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
1035 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
|
| D | stream.cc | 347 const DeviceMemory<float> &side_input, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
398 const DeviceMemory<float> &side_input, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3452 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3475 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3500 const DeviceMemory<U>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3526 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3546 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3568 const DeviceMemory<U>& 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 | 1745 auto& side_input = op_context.op_info.inputs(3); in PredictFusedConv2DBiasActivation() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3573 TRT_ShapedWeights side_input = inputs.at(3).weights(); in ConvertFusedConv2DBiasActivation() local
|