| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fused_batch_norm_ex_op_test.cc | 159 Output side_input = ops::Const(root.WithOpName("side_input"), in RunFusedBatchNorm() local
250 Output side_input = ops::Const(root.WithOpName("side_input"), in RunFusedBatchNormEx() local
385 Tensor side_input(t_dtype, input_shape); in VerifyTensorsNear() local
|
| H A 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()
|
| H A D | fused_batch_norm_op.cc | 1275 const Tensor* side_input = has_side_input_ ? &context->input(5) : nullptr; in ComputeWithReservedSpace() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | gpu_fusion.cc | 72 Value side_input; in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_conv_runner.cc | 124 se::DeviceMemory<OutputType> side_input(params.fusion->side_input_buf); in RunGpuConvForwardActivation() local
|
| H A D | cudnn_fused_conv_rewriter.cc | 347 HloInstruction* side_input; in FuseSideInputAlpha() local
|
| H A D | jitrt_custom_calls.cc | 721 Optional<SideInputAttrs> side_input = llvm::None) { in GetConvDescriptor()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | batch_mm.cpp | 331 at::Tensor side_input; in __anon957c7f480602() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | mkl_remapper_test.cc | 342 auto side_input = in TEST_F() local
|
| H A D | remapper_test.cc | 383 auto side_input = Placeholder(s.WithOpName("side_input"), DT_FLOAT, in TEST_F() local
480 auto side_input = Placeholder(s.WithOpName("side_input"), DT_FLOAT, in TEST_F() local
|
| H A D | remapper.cc | 119 int side_input = kMissingIndex; member
2802 const NodeDef& side_input = graph->node(matched.side_input); in AddFusedBatchNormExNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 1167 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
1186 const DeviceMemory<Eigen::half>& side_input, in DoBatchNormalizationForward()
|
| H A D | stream.cc | 333 const DeviceMemory<float> &side_input, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
386 const DeviceMemory<Eigen::half> &side_input, in ThenBatchNormalizationForward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 3641 const DeviceMemory<Eigen::half>& side_input, in DoBatchNormalizationForward()
3662 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3684 const DeviceMemory<T>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForwardImpl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 5413 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
5436 const DeviceMemory<Eigen::half>& side_input, in DoBatchNormalizationForward()
5462 const DeviceMemory<T>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForwardImpl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | op_level_cost_estimator.cc | 1778 auto& side_input = op_context.op_info.inputs(3); in PredictFusedConv2DBiasActivation() local
|
| H A D | op_level_cost_estimator_test.cc | 281 auto side_input = op_context.op_info.add_inputs(); in DescribeFusedConv2DBiasActivation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 3139 TRT_ShapedWeights side_input = inputs.at(3).weights(); in ConvertFusedConv2DBiasActivation() local
|