Searched refs:operand0_shape (Results 1 – 3 of 3) sorted by relevance
262 const Shape& operand0_shape = desc.operand0_shape; in GetGpuConvConfig() local267 config.input_type = operand0_shape.element_type(); in GetGpuConvConfig()283 config.input_shape = operand0_shape; in GetGpuConvConfig()290 config.output_shape = operand0_shape; in GetGpuConvConfig()293 config.input_shape = operand0_shape; in GetGpuConvConfig()447 descriptor.operand0_shape = cudnn_call->operand(0)->shape(); in GetGpuConvConfig()
128 Shape operand0_shape; member
1146 descriptor.operand0_shape = in EmitConvolutionThunkFromMlir()1169 dim->set_size(descriptor.operand0_shape.dimensions(kernel_dim)); in EmitConvolutionThunkFromMlir()