Searched refs:operand1_shape (Results 1 – 5 of 5) sorted by relevance
293 const Shape& operand1_shape = desc.operand1_shape; in GetGpuConvConfig() local307 config.filter_shape = operand1_shape; in GetGpuConvConfig()312 config.filter_shape = operand1_shape; in GetGpuConvConfig()318 config.output_shape = operand1_shape; in GetGpuConvConfig()470 descriptor.operand1_shape = cudnn_call->operand(1)->shape(); in GetGpuConvConfig()
191 Shape operand1_shape; member
735 descriptor.operand1_shape = apply_layout(operand1, b.operand_1_layout); in GetConvDescriptor()
1009 descriptor.operand1_shape = in EmitConvolutionThunk()
1207 Shape operand1_shape = ShapeUtil::MakeShape(S32, {3, 3}); in TEST_F() local1213 auto operand1 = Parameter(&builder, 1, operand1_shape, "operand1"); in TEST_F()