Searched defs:side_input (Results 1 – 10 of 10) sorted by relevance
17 TensorProto side_input = 6; field
37 side_input=None, argument
672 side_input, biases, apply_relu): argument
37 HloInstruction* side_input; member72 HloInstruction* side_input = nullptr; in FindConvWithRelu() local
281 se::DeviceMemory<T> side_input(params.fusion->side_input_buf); in RunCudnnConvImpl() local
82 const Tensor& bias, const Tensor* side_input, in LogFusedConvAutotuneResults()
196 const Tensor& side_input = context->input(kSideInput); in Compute() local314 const Tensor& bias, const Tensor* side_input, in LogFusedConvAutotuneResults()544 const Tensor* side_input = &side_input_param; in launch() local
168 def f(x, side_input): argument
225 auto side_input = op_context.op_info.add_inputs(); in DescribeFusedConv2DBiasActivation() local
1177 auto& side_input = op_context.op_info.inputs(3); in PredictFusedConv2DBiasActivation() local