Searched refs:kernel_tensor_shape (Results 1 – 1 of 1) sorted by relevance
1621 auto kernel_tensor_shape = in PrecheckForXlaConvV2Op() local1631 if (kernel_tensor_shape.size() <= 2) { in PrecheckForXlaConvV2Op()1633 << "kernel tensor argument is " << kernel_tensor_shape.size() in PrecheckForXlaConvV2Op()1638 if (input_tensor_shape.size() != kernel_tensor_shape.size()) { in PrecheckForXlaConvV2Op()1667 kernel_tensor_shape[dnums.kernel_input_feature_dimension()]; in PrecheckForXlaConvV2Op()1669 kernel_tensor_shape[dnums.kernel_output_feature_dimension()]; in PrecheckForXlaConvV2Op()1756 auto kernel_tensor_shape = kernel_tensor.getType().cast<RankedTensorType>(); in InferShapeForXlaConvV2Op() local1757 for (auto i = 0; i < kernel_tensor_shape.getShape().size(); ++i) { in InferShapeForXlaConvV2Op()1759 << kernel_tensor_shape.getShape()[i]); in InferShapeForXlaConvV2Op()1760 kernel_tensor_dims_vec.push_back(kernel_tensor_shape.getShape()[i]); in InferShapeForXlaConvV2Op()