Searched defs:conv_input (Results 1 – 5 of 5) sorted by relevance
30 def fused_conv2d_bias_activation(conv_input, argument
670 def _SimulateFusedConv2dBiasActivationInt8(conv_input_scale, conv_input, kernel, argument
174 const Tensor& conv_input = context->input(kConvInput); in Compute() local483 const Tensor* conv_input = &conv_input_param; in launch() local
246 xla::XlaOp conv_input, in MakeXlaForwardConvOp()
1174 auto& conv_input = op_context.op_info.inputs(0); in PredictFusedConv2DBiasActivation() local